Skip to content

Commit c1129e6

Browse files
gs-olivegs-olive
committed
fix: Reorganize testing directories
1 parent c5975fb commit c1129e6

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

44 files changed

+104
-56
lines changed

‎.circleci/config.yml

+15
Original file line numberDiff line numberDiff line change
@@ -768,6 +768,20 @@ commands:
768768
- store_artifacts:
769769
path: /tmp/testlogs
770770

771+
test-dynamo-shared_utilities:
772+
description: "Test Dynamo shared utilities tests"
773+
steps:
774+
- run:
775+
name: Run Dynamo shared_utilities tests
776+
command: |
777+
cd tests/py/dynamo/shared_utilities/
778+
pytest --junitxml=/tmp/artifacts/test_results/dynamo/shared_utilities/test_results.xml
779+
780+
- store_test_results:
781+
path: /tmp/artifacts
782+
- store_artifacts:
783+
path: /tmp/testlogs
784+
771785
test-dynamo-models_torch_compile:
772786
description: "Test the Dynamo models via torch_compile path"
773787
steps:
@@ -1080,6 +1094,7 @@ jobs:
10801094
- dump-test-env
10811095
- test-dynamo-converters
10821096
- test-dynamo-torch_compile
1097+
- test-dynamo-shared_utilities
10831098
- test-dynamo-models_torch_compile
10841099
- test-dynamo-models_torch_export
10851100

‎tests/py/dynamo/__init__.py

Whitespace-only changes.

‎tests/py/dynamo/backend/__init__.py

Whitespace-only changes.

‎tests/py/dynamo/backend/test_backend_compiler.py

+5-3
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,11 @@
1+
from copy import deepcopy
2+
13
import torch
24
import torch_tensorrt
5+
from torch.testing._internal.common_utils import TestCase, run_tests
36
from torch_tensorrt.dynamo.partitioning import fast_partition
4-
from torch.testing._internal.common_utils import run_tests, TestCase
5-
from copy import deepcopy
6-
from utils import lower_graph_testing, DECIMALS_OF_AGREEMENT
7+
8+
from ..testing_utilities import DECIMALS_OF_AGREEMENT, lower_graph_testing
79

810

911
class TestTRTModuleNextCompilation(TestCase):

‎tests/py/dynamo/backend/test_pre_aot_lowering.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,8 @@
11
import torch
22
import torch_tensorrt
3-
from utils import lower_graph_testing
4-
from torch.testing._internal.common_utils import run_tests, TestCase
3+
from torch.testing._internal.common_utils import TestCase, run_tests
4+
5+
from ..testing_utilities import lower_graph_testing
56

67

78
class TestMaxPool1D(TestCase):

‎tests/py/dynamo/backend/test_specialized_models.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,8 @@
1-
from utils import lower_graph_testing
2-
from torch.testing._internal.common_utils import run_tests, TestCase
31
import torch
42
import torch_tensorrt
3+
from torch.testing._internal.common_utils import TestCase, run_tests
4+
5+
from ..testing_utilities import lower_graph_testing
56

67

78
class TestFakeTensors(TestCase):

‎tests/py/dynamo/converters/__init__.py

Whitespace-only changes.

‎tests/py/dynamo/converters/test_adaptive_avgpool_aten.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
from torch.testing._internal.common_utils import run_tests
44
from torch_tensorrt import Input
55

6-
from harness import DispatchTestCase
6+
from .harness import DispatchTestCase
77

88

99
class TestAdaptiveAvgPoolConverter(DispatchTestCase):

‎tests/py/dynamo/converters/test_batchnorm_aten.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,9 @@
11
import torch
22
from torch.testing._internal.common_utils import run_tests
3-
from harness import DispatchTestCase
43
from torch_tensorrt import Input
54

5+
from .harness import DispatchTestCase
6+
67

78
class TestBatchNormConverter(DispatchTestCase):
89
def test_batchnorm(self):

‎tests/py/dynamo/converters/test_binary_ops_aten.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,14 @@
1-
from typing import Callable
21
import unittest
2+
from typing import Callable
33

44
import torch
55
import torch.nn as nn
6-
76
from parameterized import parameterized
87
from torch.testing._internal.common_utils import run_tests
9-
from harness import DispatchTestCase
108
from torch_tensorrt import Input
119

10+
from .harness import DispatchTestCase
11+
1212
NEED_TEST_BOTH_CONSTANTS_CASE = True
1313

1414
elementwise_ops = [

‎tests/py/dynamo/converters/test_casts.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
11
import torch
22
import torch.nn as nn
3-
from harness import DispatchTestCase
43
from torch.testing._internal.common_utils import run_tests
54
from torch_tensorrt.dynamo.conversion import UnsupportedOperatorException
65

6+
from .harness import DispatchTestCase
7+
78

89
class TestCloneConverter(DispatchTestCase):
910
def test_clone_contiguous(self):

‎tests/py/dynamo/converters/test_cat_aten.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,10 @@
22
import torch.nn as nn
33
from parameterized import parameterized
44
from torch.testing._internal.common_utils import run_tests
5-
from harness import DispatchTestCase
65
from torch_tensorrt import Input
76

7+
from .harness import DispatchTestCase
8+
89

910
class TestCatConverter(DispatchTestCase):
1011
@parameterized.expand(

‎tests/py/dynamo/converters/test_clamp_aten.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
11
import torch
22
from parameterized import param, parameterized
33
from torch.testing._internal.common_utils import run_tests
4-
from harness import DispatchTestCase
54
from torch_tensorrt import Input
65

6+
from .harness import DispatchTestCase
7+
78

89
class TestClampConverter(DispatchTestCase):
910
@parameterized.expand(

‎tests/py/dynamo/converters/test_convolution_aten.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
11
import torch
22
from parameterized import param, parameterized
33
from torch.testing._internal.common_utils import run_tests
4-
from harness import DispatchTestCase
54
from torch_tensorrt import Input
65

6+
from .harness import DispatchTestCase
7+
78

89
class TestConvolutionConverter(DispatchTestCase):
910
@parameterized.expand(

‎tests/py/dynamo/converters/test_elu_aten.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
11
import torch
22
import torch.nn as nn
33
from torch.testing._internal.common_utils import run_tests
4-
from harness import DispatchTestCase
54
from torch_tensorrt import Input
65

6+
from .harness import DispatchTestCase
7+
78

89
class TestELUConverter(DispatchTestCase):
910
def test_elu(self):

‎tests/py/dynamo/converters/test_embedding_aten.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,11 @@
11
import torch
22
import torch.nn as nn
3-
from torch.testing._internal.common_utils import run_tests
4-
from harness import DispatchTestCase
53
from parameterized import param, parameterized
4+
from torch.testing._internal.common_utils import run_tests
65
from torch_tensorrt import Input
76

7+
from .harness import DispatchTestCase
8+
89

910
class TestEmbeddingConverter(DispatchTestCase):
1011
@parameterized.expand(

‎tests/py/dynamo/converters/test_evaluators.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,10 @@
33

44
import torch
55
import torch.nn as nn
6-
from harness import DispatchTestCase
76
from torch.testing._internal.common_utils import run_tests
87

8+
from .harness import DispatchTestCase
9+
910

1011
# TODO: Switch this test back to self.run_test once an implementation exists
1112
# for a converter that returns a list, such as aten.split

‎tests/py/dynamo/converters/test_expand_aten.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,8 @@
22
import torch.nn as nn
33
from parameterized import parameterized
44
from torch.testing._internal.common_utils import run_tests
5-
from harness import DispatchTestCase
5+
6+
from .harness import DispatchTestCase
67

78

89
class TestExpandConverter(DispatchTestCase):

‎tests/py/dynamo/converters/test_gelu_aten.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
11
import torch
22
import torch.nn as nn
33
from torch.testing._internal.common_utils import run_tests
4-
from harness import DispatchTestCase
54
from torch_tensorrt import Input
65

6+
from .harness import DispatchTestCase
7+
78

89
class TestGeLUConverter(DispatchTestCase):
910
def test_gelu(self):

‎tests/py/dynamo/converters/test_hardtanh_aten.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
11
import torch
22
import torch.nn as nn
33
from torch.testing._internal.common_utils import run_tests
4-
from harness import DispatchTestCase
54
from torch_tensorrt import Input
65

6+
from .harness import DispatchTestCase
7+
78

89
class TestHardTanHConverter(DispatchTestCase):
910
def test_hardtanh(self):

‎tests/py/dynamo/converters/test_layer_norm_aten.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,9 @@
11
import torch
22
from torch.testing._internal.common_utils import run_tests
3-
from harness import DispatchTestCase
43
from torch_tensorrt import Input
54

5+
from .harness import DispatchTestCase
6+
67

78
class TestLayerNormConverter(DispatchTestCase):
89
def test_layer_norm(self):

‎tests/py/dynamo/converters/test_leaky_relu_aten.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
11
import torch
22
import torch.nn as nn
33
from torch.testing._internal.common_utils import run_tests
4-
from harness import DispatchTestCase
54
from torch_tensorrt import Input
65

6+
from .harness import DispatchTestCase
7+
78

89
class TestLeakyReLUConverter(DispatchTestCase):
910
def test_leaky_relu(self):

‎tests/py/dynamo/converters/test_linear_aten.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,8 @@
11
import torch
22
from parameterized import parameterized
33
from torch.testing._internal.common_utils import run_tests
4-
from harness import DispatchTestCase
4+
5+
from .harness import DispatchTestCase
56

67

78
class TestLinearConverter(DispatchTestCase):

‎tests/py/dynamo/converters/test_matmul_aten.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,8 @@
22
import torch.nn as nn
33
from parameterized import parameterized
44
from torch.testing._internal.common_utils import run_tests
5-
from harness import DispatchTestCase
5+
6+
from .harness import DispatchTestCase
67

78

89
class TestMatMulConverter(DispatchTestCase):

‎tests/py/dynamo/converters/test_mean_aten.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
11
import torch
22
import torch.nn as nn
33
from torch.testing._internal.common_utils import run_tests
4-
from harness import DispatchTestCase
54
from torch_tensorrt import Input
65

6+
from .harness import DispatchTestCase
7+
78

89
class TestMeanDimConverter(DispatchTestCase):
910
def test_mean_dim_keepdims(self):

‎tests/py/dynamo/converters/test_permutation_aten.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,10 @@
22
import torch.nn as nn
33
from parameterized import parameterized
44
from torch.testing._internal.common_utils import run_tests
5-
from harness import DispatchTestCase
65
from torch_tensorrt import Input
76

7+
from .harness import DispatchTestCase
8+
89

910
class TestPermuteConverter(DispatchTestCase):
1011
@parameterized.expand(

‎tests/py/dynamo/converters/test_relu_aten.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
11
import torch
22
import torch.nn as nn
33
from torch.testing._internal.common_utils import run_tests
4-
from harness import DispatchTestCase
54
from torch_tensorrt import Input
65

6+
from .harness import DispatchTestCase
7+
78

89
class TestReLUConverter(DispatchTestCase):
910
def test_relu(self):

‎tests/py/dynamo/converters/test_reshape_aten.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,10 @@
44
import torch
55
from parameterized import parameterized
66
from torch.testing._internal.common_utils import run_tests
7-
from harness import DispatchTestCase
87
from torch_tensorrt import Input
98

9+
from .harness import DispatchTestCase
10+
1011

1112
class TestReshapeConverter(DispatchTestCase):
1213
@parameterized.expand(

‎tests/py/dynamo/converters/test_rsqrt_aten.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,10 @@
22
import torch.nn as nn
33
from parameterized import parameterized
44
from torch.testing._internal.common_utils import run_tests
5-
from harness import DispatchTestCase
65
from torch_tensorrt import Input
76

7+
from .harness import DispatchTestCase
8+
89

910
class TestRSqrtConverter(DispatchTestCase):
1011
@parameterized.expand(

‎tests/py/dynamo/converters/test_select_aten.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
11
import torch
22
from parameterized import parameterized
33
from torch.testing._internal.common_utils import run_tests
4-
from harness import DispatchTestCase
54
from torch_tensorrt import Input
65

6+
from .harness import DispatchTestCase
7+
78

89
class TestSelectConverterOne(DispatchTestCase):
910
@parameterized.expand(

‎tests/py/dynamo/converters/test_selu_aten.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
11
import torch
22
import torch.nn as nn
33
from torch.testing._internal.common_utils import run_tests
4-
from harness import DispatchTestCase
54
from torch_tensorrt import Input
65

6+
from .harness import DispatchTestCase
7+
78

89
class TestSeLUConverter(DispatchTestCase):
910
def test_selu(self):

‎tests/py/dynamo/converters/test_sigmoid_aten.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
11
import torch
22
import torch.nn as nn
33
from torch.testing._internal.common_utils import run_tests
4-
from harness import DispatchTestCase
54
from torch_tensorrt import Input
65

6+
from .harness import DispatchTestCase
7+
78

89
class TestSigmoidConverter(DispatchTestCase):
910
def test_sigmoid(self):

‎tests/py/dynamo/converters/test_slice_aten.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
11
import torch
22
from parameterized import parameterized
33
from torch.testing._internal.common_utils import run_tests
4-
from harness import DispatchTestCase
54
from torch_tensorrt import Input
65

6+
from .harness import DispatchTestCase
7+
78

89
class TestSelectConverterImplicitBatch(DispatchTestCase):
910
@parameterized.expand(

‎tests/py/dynamo/converters/test_softmax_aten.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,9 @@
11
import torch
22
from torch.testing._internal.common_utils import run_tests
3-
from harness import DispatchTestCase
43
from torch_tensorrt import Input
54

5+
from .harness import DispatchTestCase
6+
67

78
class TestSoftMaxConverter(DispatchTestCase):
89
def test_softmax(self):

‎tests/py/dynamo/converters/test_squeeze_aten.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,10 @@
22
import torch.nn as nn
33
from parameterized import parameterized
44
from torch.testing._internal.common_utils import run_tests
5-
from harness import DispatchTestCase
65
from torch_tensorrt import Input
76

7+
from .harness import DispatchTestCase
8+
89

910
class TestSqueezeConverter(DispatchTestCase):
1011
@parameterized.expand(

0 commit comments

Comments
 (0)