Skip to content

Commit a878b78

Browse files
committed
disabling dla args for hope igx platform
1 parent 5e38c04 commit a878b78

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

tests/py/dynamo/runtime/test_000_compilation_settings.py

Lines changed: 9 additions & 0 deletions
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,11 @@ def forward(self, x):
5357
)
5458
torch._dynamo.reset()
5559

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

0 commit comments

Comments
 (0)