[TRT] fix outdated module building method in tensorrt (#16031)

apache · Nov 6, 2023 · 7294a1e · 7294a1e
1 parent 1de5aa5
commit 7294a1e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/how_to/deploy/tensorrt.rst b/docs/how_to/deploy/tensorrt.rst
@@ -96,7 +96,7 @@ regular TVM CUDA compilation and code generation.
 .. code:: python
 
     from tvm.relay.op.contrib.tensorrt import partition_for_tensorrt
-    mod, config = partition_for_tensorrt(mod, params)
+    mod = partition_for_tensorrt(mod, params)
 
 
 Build the Relay graph, using the new module and config returned by partition_for_tensorrt. The
@@ -107,7 +107,7 @@ PassContext so the values can be read during compilation.
 .. code:: python
 
     target = "cuda"
-    with tvm.transform.PassContext(opt_level=3, config={'relay.ext.tensorrt.options': config}):
+    with tvm.transform.PassContext(opt_level=3):
         lib = relay.build(mod, target=target, params=params)