Renamed convert_exported_program_to_serialized_trt_engine (#3066)

pytorch · Aug 6, 2024 · 18ab1e6 · 18ab1e6
1 parent 2206a6b
commit 18ab1e6
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/py/dynamo/models/test_export_kwargs_serde.py b/tests/py/dynamo/models/test_export_kwargs_serde.py
@@ -10,7 +10,9 @@
 import torch_tensorrt as torchtrt
 import torchvision.models as models
 from torch import nn
-from torch_tensorrt.dynamo._compiler import convert_module_to_trt_engine
+from torch_tensorrt.dynamo._compiler import (
+    convert_exported_program_to_serialized_trt_engine,
+)
 from torch_tensorrt.dynamo.utils import (
     COSINE_THRESHOLD,
     cosine_similarity,
@@ -507,4 +509,6 @@ def forward(self, x, b=5, c=None, d=None):
     }
 
     exp_program = torch.export.export(model, args=tuple(args), kwargs=kwargs)
-    engine = convert_module_to_trt_engine(exp_program, **compile_spec)
+    engine = convert_exported_program_to_serialized_trt_engine(
+        exp_program, **compile_spec
+    )