fix: fix exported_program import error (#3007)

peri044 · web-flow · commit a835d5fd26df · 2024-07-17T11:41:55.000-07:00
diff --git a/py/torch_tensorrt/_compile.py b/py/torch_tensorrt/_compile.py
@@ -24,7 +24,7 @@
     )
 
 if ENABLED_FEATURES.dynamo_frontend:
-    from torch._export import ExportedProgram
+    from torch.export import ExportedProgram
     from torch_tensorrt.dynamo._compiler import compile as dynamo_compile
     from torch_tensorrt.dynamo._compiler import (
         convert_module_to_trt_engine as dynamo_convert_module_to_trt_engine,

Original file line number	Diff line number	Diff line change
`@@ -24,7 +24,7 @@`
`24`	`24`	`)`
`25`	`25`
`26`	`26`	`if ENABLED_FEATURES.dynamo_frontend:`
`27`		`- from torch._export import ExportedProgram`
	`27`	`+ from torch.export import ExportedProgram`
`28`	`28`	`from torch_tensorrt.dynamo._compiler import compile as dynamo_compile`
`29`	`29`	`from torch_tensorrt.dynamo._compiler import (`
`30`	`30`	`convert_module_to_trt_engine as dynamo_convert_module_to_trt_engine,`