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
1
2
from torch_tensorrt .dynamo .runtime ._PythonTorchTensorRTModule import ( # noqa: F401
2
3
PythonTorchTensorRTModule ,
3
4
)
4
5
from torch_tensorrt .dynamo .runtime ._TorchTensorRTModule import ( # noqa: F401
5
6
TorchTensorRTModule ,
6
7
)
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):
593
593
)
594
594
]
595
595
596
- packages += [
597
- "torch_tensorrt.ts" ,
598
- ]
596
+ packages += ["torch_tensorrt.ts" , "torch_tensorrt.dynamo.runtime.meta_ops" ]
599
597
600
598
package_dir .update (
601
599
{
602
600
"torch_tensorrt.ts" : "py/torch_tensorrt/ts" ,
601
+ "torch_tensorrt.dynamo.runtime.meta_ops" : "py/torch_tensorrt/dynamo/runtime/meta_ops" ,
603
602
}
604
603
)
605
604
You can’t perform that action at this time.
0 commit comments