41 lines
1.4 KiB
Python
41 lines
1.4 KiB
Python
# isort: skip_file
|
|
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
# pylint: disable=unused-import
|
|
"""Common data structures across all IR variants."""
|
|
|
|
from . import instrument, transform
|
|
from .attrs import Attrs, DictAttrs, make_node
|
|
from .base import (
|
|
EnvFunc,
|
|
Node,
|
|
SourceName,
|
|
Span,
|
|
SequentialSpan,
|
|
assert_structural_equal,
|
|
load_json,
|
|
save_json,
|
|
)
|
|
from .expr import Call, Expr, GlobalVar, Range, is_prim_expr
|
|
from .function import BaseFunc, CallingConv
|
|
from .global_info import GlobalInfo, DummyGlobalInfo, VDevice
|
|
from .module import IRModule
|
|
from .op import Op, register_intrin_lowering, register_op_attr
|
|
from .type import FuncType, PointerType, PrimType, TupleType, Type
|
|
|
|
from tvm_ffi import Array, Map
|