Skip to content

Commit 519f75d

Browse files
author
pytorchbot
committed
2025-05-02 nightly release (0da76f3)
1 parent 77d2338 commit 519f75d

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

tests/py/dynamo/runtime/test_000_compilation_settings.py

+8
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,10 @@
1+
import unittest
2+
3+
import tensorrt as trt
14
import torch
25
import torch_tensorrt
36
from torch.testing._internal.common_utils import TestCase, run_tests
7+
from torch_tensorrt.dynamo.utils import is_tegra_platform
48

59
from ..testing_utilities import DECIMALS_OF_AGREEMENT
610

@@ -53,6 +57,10 @@ def forward(self, x):
5357
)
5458
torch._dynamo.reset()
5559

60+
@unittest.skipIf(
61+
is_tegra_platform() and trt._version_ > "10.8",
62+
"DLA is not supported on Jetson platform starting TRT 10.8",
63+
)
5664
def test_dla_args(self):
5765
class AddSoftmax(torch.nn.Module):
5866
def forward(self, x):

0 commit comments

Comments
 (0)