Skip to content

Commit fcbfcb4

Browse files
authored
Jetson distributed fix (#3716)
1 parent 3876b09 commit fcbfcb4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

py/torch_tensorrt/dynamo/backend/backends.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ def aot_torch_tensorrt_aten_backend(
8181
decompositions=settings_aot_autograd["decompositions"],
8282
)(gm, sample_inputs)
8383

84-
if is_tegra_platform():
84+
if not is_tegra_platform():
8585
from torch.distributed.tensor import DTensor
8686

8787
if any(isinstance(tensor, DTensor) for tensor in sample_inputs):

0 commit comments

Comments
 (0)