File tree Expand file tree Collapse file tree 3 files changed +6
-4
lines changed
py/torch_tensorrt/dynamo/runtime Expand file tree Collapse file tree 3 files changed +6
-4
lines changed Original file line number Diff line number Diff line change 1+ import torch_tensorrt
12from torch_tensorrt .dynamo .runtime ._PythonTorchTensorRTModule import ( # noqa: F401
23 PythonTorchTensorRTModule ,
34)
45from torch_tensorrt .dynamo .runtime ._TorchTensorRTModule import ( # noqa: F401
56 TorchTensorRTModule ,
67)
7- from torch_tensorrt .dynamo .runtime .register_fake_class import *
8+
9+ if torch_tensorrt .ENABLED_FEATURES .torch_tensorrt_runtime :
10+ from torch_tensorrt .dynamo .runtime .meta_ops .register_meta_ops import *
File renamed without changes.
Original file line number Diff line number Diff line change @@ -593,13 +593,12 @@ def run(self):
593593 )
594594 ]
595595
596- packages += [
597- "torch_tensorrt.ts" ,
598- ]
596+ packages += ["torch_tensorrt.ts" , "torch_tensorrt.dynamo.runtime.meta_ops" ]
599597
600598 package_dir .update (
601599 {
602600 "torch_tensorrt.ts" : "py/torch_tensorrt/ts" ,
601+ "torch_tensorrt.dynamo.runtime.meta_ops" : "py/torch_tensorrt/dynamo/runtime/meta_ops" ,
603602 }
604603 )
605604
You can’t perform that action at this time.
0 commit comments