From db505234d816c64038135df41bf92143bc3a4b13 Mon Sep 17 00:00:00 2001 From: Siyuan Feng Date: Sun, 28 Jan 2024 19:00:07 +0800 Subject: [PATCH] [Doc] Fix Docstring in `extern.py` for Sphinx 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 8c0491eaec90..332d07cbc3c5 100644 --- a/python/tvm/relax/frontend/nn/extern.py +++ b/python/tvm/relax/frontend/nn/extern.py @@ -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 #include @@ -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`