Skip to content

Commit

Permalink
[Doc] Fix Docstring in extern.py for Sphinx (apache#16482)
Browse files Browse the repository at this point in the history
Fix a missing line in the docstring of `extern.py`, which would cause
Sphinx to fail to build the documentation.
  • Loading branch information
Hzfengsy authored Jan 28, 2024
1 parent a15fd1d commit b4fc014
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/tvm/relax/frontend/nn/extern.py
Original file line number Diff line number Diff line change
Expand Up @@ -138,6 +138,7 @@ def shape_dtype_inference(a, b):
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>
Expand All @@ -152,7 +153,6 @@ def shape_dtype_inference(a, b):
// 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`
Expand Down

0 comments on commit b4fc014

Please sign in to comment.