Skip to content

Commit 78d4ecd

Browse files
committed
Mark FX2TRT converter as fake tensor unsupported
1 parent 4ed0992 commit 78d4ecd

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

py/torch_tensorrt/fx/tracer/dispatch_tracer/tensorrt_dynamo_backend.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,8 @@
1515
from torch_tensorrt.fx.trt_module import TRTModule
1616
from torch_tensorrt.fx.utils import LowerPrecision
1717

18+
from torch._dynamo.backends.common import fake_tensor_unsupported
19+
1820
from torch._functorch.aot_autograd import aot_module_simplified, make_boxed_compiler
1921

2022
from torch._inductor.decomposition import decompositions
@@ -99,6 +101,7 @@ def get_input(self, inputs):
99101

100102

101103
@td.register_backend
104+
@fake_tensor_unsupported
102105
def fx2trt_compiler(gm: torch.fx.GraphModule, example_inputs):
103106
try:
104107
trt_compiled = fx2trt(gm, example_inputs)

0 commit comments

Comments
 (0)