enable prune_tensorrt_subgraphs

Jian Sheng · Jian Sheng · commit 02ae8442eb04 · 2022-07-18T21:00:55.000Z
diff --git a/python/tvm/relay/op/contrib/tensorrt.py b/python/tvm/relay/op/contrib/tensorrt.py
@@ -177,7 +177,7 @@ def partition_for_tensorrt(
     with tvm.transform.PassContext(opt_level=3, config={"relay.ext.tensorrt.options": config}):
         mod = seq(mod)
         # TODO(mbs): Revisit
-        # mod = prune_tensorrt_subgraphs(mod)
+        mod = prune_tensorrt_subgraphs(mod)
     return mod, config
 
 
