dev.rst 1.32 KB
Newer Older
1 2 3
Developer API
-------------
This page contains modules that are used by developers of TVM.
4
Many of these APIs are PackedFunc registered in C++ backend.
5 6 7

tvm.node
~~~~~~~~
8
.. automodule:: tvm.node
9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47

.. autoclass:: tvm.node.NodeBase
    :members:

.. autoclass:: tvm.node.Node
    :members:

.. autofunction:: tvm.register_node

tvm.expr
~~~~~~~~
.. automodule:: tvm.expr
   :members:
   :undoc-members:

tvm.codegen
~~~~~~~~~~~
.. automodule:: tvm.codegen
   :members:
   :undoc-members:

tvm.stmt
~~~~~~~~
.. automodule:: tvm.stmt
   :members:
   :undoc-members:

tvm.ir_pass
~~~~~~~~~~~
.. autosummary::

   tvm.ir_pass.Inline
   tvm.ir_pass.Simplify
   tvm.ir_pass.ConvertSSA
   tvm.ir_pass.VerifySSA
   tvm.ir_pass.CanonicalSimplify
   tvm.ir_pass.StorageFlatten
   tvm.ir_pass.VectorizeLoop
   tvm.ir_pass.UnrollLoop
48
   tvm.ir_pass.ThreadSync
49
   tvm.ir_pass.StorageRewrite
50 51 52 53 54 55 56
   tvm.ir_pass.MakeAPI
   tvm.ir_pass.SplitHostDevice
   tvm.ir_pass.InjectVirtualThread
   tvm.ir_pass.LoopPartition
   tvm.ir_pass.RemoveNoOp
   tvm.ir_pass.SplitPipeline
   tvm.ir_pass.LowerThreadAllreduce
57
   tvm.ir_pass.LowerIntrin
58
   tvm.ir_pass.LowerTVMBuiltin
59 60
   tvm.ir_pass.NarrowChannelAccess

61 62 63 64 65 66 67
.. automodule:: tvm.ir_pass
   :members:

tvm.ir_builder
~~~~~~~~~~~~~~
.. automodule:: tvm.ir_builder
   :members:
68 69 70 71 72

tvm.make
~~~~~~~~
.. automodule:: tvm.make
   :members: