From 7459f8c40e9f95b298a4addd06afc40fbe1a1580 Mon Sep 17 00:00:00 2001 From: Junho Wohn Date: Wed, 1 Nov 2023 18:34:45 +0900 Subject: [PATCH] [TRT] fix outdated module building method in tensorrt --- docs/how_to/deploy/tensorrt.rst | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/how_to/deploy/tensorrt.rst b/docs/how_to/deploy/tensorrt.rst index b91f7b1a874a..cb1b309db0f0 100644 --- 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)