Skip to content

Commit

Permalink
[REFACTOR][PY] Establish tvm.tir
Browse files Browse the repository at this point in the history
- Move related files into the corresponding location as in C++
- Keep the top-level TVM API backward compatible to make minimum changes in topi
  • Loading branch information
tqchen authored and zhiics committed Mar 2, 2020
1 parent 16c341b commit 7c3fd2e
Show file tree
Hide file tree
Showing 121 changed files with 2,717 additions and 2,481 deletions.
24 changes: 14 additions & 10 deletions python/tvm/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,36 +33,40 @@
from .runtime.ndarray import vpi, rocm, opengl, ext_dev, micro_dev
from .runtime import ndarray as nd

# tvm.error
from . import error

# tvm.ir
from .ir import IRModule
from .ir import transform
from .ir import container
from . import ir

# tvm.tir
from . import tir

# tvm.target
from . import target

# others
from . import tensor
from . import arith
from . import expr
from . import stmt
from . import make
from . import ir_pass
from . import schedule

from . import ir_builder
from . import target
from . import generic
from . import hybrid
from . import testing
from . import error


from .api import *
from .intrin import *
from .tensor_intrin import decl_tensor_intrin
from .schedule import create_schedule
from .build_module import build, lower, build_config
from .tag import tag_scope

# backward compact for topi, to be removed later
from .tir import expr, stmt, ir_builder, ir_pass, generic
from .tir.op import *
from . import intrin

# Contrib initializers
from .contrib import rocm as _rocm, nvcc as _nvcc, sdaccel as _sdaccel

Expand Down
Loading

0 comments on commit 7c3fd2e

Please sign in to comment.