Skip to content

Commit 18ab1e6

Browse files
authored
Renamed convert_exported_program_to_serialized_trt_engine (#3066)
1 parent 2206a6b commit 18ab1e6

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

tests/py/dynamo/models/test_export_kwargs_serde.py

+6-2
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,9 @@
1010
import torch_tensorrt as torchtrt
1111
import torchvision.models as models
1212
from torch import nn
13-
from torch_tensorrt.dynamo._compiler import convert_module_to_trt_engine
13+
from torch_tensorrt.dynamo._compiler import (
14+
convert_exported_program_to_serialized_trt_engine,
15+
)
1416
from torch_tensorrt.dynamo.utils import (
1517
COSINE_THRESHOLD,
1618
cosine_similarity,
@@ -507,4 +509,6 @@ def forward(self, x, b=5, c=None, d=None):
507509
}
508510

509511
exp_program = torch.export.export(model, args=tuple(args), kwargs=kwargs)
510-
engine = convert_module_to_trt_engine(exp_program, **compile_spec)
512+
engine = convert_exported_program_to_serialized_trt_engine(
513+
exp_program, **compile_spec
514+
)

0 commit comments

Comments
 (0)