Skip to content

Commit 99019f6

Browse files
leeexyzAWS Neo
authored andcommitted
[DOC]Update doc in _api_internal.py and ir_pass.py (apache#2514)
1 parent 01ba738 commit 99019f6

File tree

2 files changed

+11
-2
lines changed

2 files changed

+11
-2
lines changed

python/tvm/_api_internal.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1,9 @@
1-
"""namespace of internal API"""
1+
"""Namespace of internal API
2+
3+
The functions in this namespace are automatically exported from C++ side via PackedFunc
4+
that is registered by "TVM_REGISTER_*" macro. This way makes calling Python functions from C++
5+
side very easily.
6+
7+
Each string starts with "_" in the "TVM_REGISTER_*" macro is an internal API. You can find
8+
all the functions in "api_lang.cc", "api_base.cc", "api_arith.cc" and "api_ir.cc" under "src/api".
9+
"""

python/tvm/ir_pass.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,8 @@
44
The functions are automatically exported from C++ side via PackedFunc.
55
66
Each api is a PackedFunc that can be called in a positional argument manner.
7-
You can read "include/tvm/pass.h" for the function signature of these functions.
7+
You can read "include/tvm/ir_pass.h" for the function signature and
8+
"src/api/api_pass.cc" for the PackedFunc's body of these functions.
89
"""
910
from ._ffi.function import _init_api
1011

0 commit comments

Comments
 (0)