You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by yo...@apache.org on 2024/01/28 23:29:11 UTC
(tvm) branch main updated: [Doc] Fix Docstring in `extern.py` for Sphinx (#16482)
This is an automated email from the ASF dual-hosted git repository.
yongwww pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new b4fc01444c [Doc] Fix Docstring in `extern.py` for Sphinx (#16482)
b4fc01444c is described below
commit b4fc01444c3df99d37bfdcbde3ea363fa47f4b43
Author: Siyuan Feng <Hz...@sjtu.edu.cn>
AuthorDate: Mon Jan 29 07:29:05 2024 +0800
[Doc] Fix Docstring in `extern.py` for Sphinx (#16482)
Fix a missing line in the docstring of `extern.py`, which would cause
Sphinx to fail to build the documentation.
---
python/tvm/relax/frontend/nn/extern.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/python/tvm/relax/frontend/nn/extern.py b/python/tvm/relax/frontend/nn/extern.py
index 8c0491eaec..332d07cbc3 100644
--- a/python/tvm/relax/frontend/nn/extern.py
+++ b/python/tvm/relax/frontend/nn/extern.py
@@ -138,6 +138,7 @@ class SourceModule(ExternModule): # pylint: disable=too-few-public-methods
To expose the symbol, `TVM_DLL_EXPORT_TYPED_FUNC(symbol, function)` is guaranteed available:
.. code-block:: C++
+
// those headers are guaranteed to be available
#include <dlpack/dlpack.h>
#include <tvm/runtime/data_type.h>
@@ -152,7 +153,6 @@ class SourceModule(ExternModule): # pylint: disable=too-few-public-methods
// expose symbol `my_func` instead of `_my_func_impl`
TVM_DLL_EXPORT_TYPED_FUNC(my_func, _my_func_impl);
-
**A compiler pass `AttachExternModules`.** It is introduced to attach a list of
`nn.ExternModule`s into an IRModule at any stage of the compilation pipeline,
and attach the compiled external modules as `runtime.Module`s into IRModule's `external_mods`