We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3876b09 commit fcbfcb4Copy full SHA for fcbfcb4
py/torch_tensorrt/dynamo/backend/backends.py
@@ -81,7 +81,7 @@ def aot_torch_tensorrt_aten_backend(
81
decompositions=settings_aot_autograd["decompositions"],
82
)(gm, sample_inputs)
83
84
- if is_tegra_platform():
+ if not is_tegra_platform():
85
from torch.distributed.tensor import DTensor
86
87
if any(isinstance(tensor, DTensor) for tensor in sample_inputs):
0 commit comments