fix: Reorganize testing directories

Author: gs-olive

.circleci/config.yml

@@ -768,6 +768,20 @@ commands:
       - store_artifacts:
           path: /tmp/testlogs
 
+  test-dynamo-shared_utilities:
+    description: "Test Dynamo shared utilities tests"
+    steps:
+      - run:
+          name: Run Dynamo shared_utilities tests
+          command: |
+            cd tests/py/dynamo/shared_utilities/
+            pytest --junitxml=/tmp/artifacts/test_results/dynamo/shared_utilities/test_results.xml
+
+      - store_test_results:
+          path: /tmp/artifacts
+      - store_artifacts:
+          path: /tmp/testlogs
+
   test-dynamo-models_torch_compile:
     description: "Test the Dynamo models via torch_compile path"
     steps:
@@ -1080,6 +1094,7 @@ jobs:
       - dump-test-env
       - test-dynamo-converters
       - test-dynamo-torch_compile
+      - test-dynamo-shared_utilities
       - test-dynamo-models_torch_compile
       - test-dynamo-models_torch_export
 

tests/py/dynamo/__init__.py

@@ -1,9 +1,11 @@
+from copy import deepcopy
+
 import torch
 import torch_tensorrt
+from torch.testing._internal.common_utils import TestCase, run_tests
 from torch_tensorrt.dynamo.partitioning import fast_partition
-from torch.testing._internal.common_utils import run_tests, TestCase
-from copy import deepcopy
-from utils import lower_graph_testing, DECIMALS_OF_AGREEMENT
+
+from ..testing_utilities import DECIMALS_OF_AGREEMENT, lower_graph_testing
 
 
 class TestTRTModuleNextCompilation(TestCase):

tests/py/dynamo/backend/__init__.py

@@ -1,7 +1,8 @@
 import torch
 import torch_tensorrt
-from utils import lower_graph_testing
-from torch.testing._internal.common_utils import run_tests, TestCase
+from torch.testing._internal.common_utils import TestCase, run_tests
+
+from ..testing_utilities import lower_graph_testing
 
 
 class TestMaxPool1D(TestCase):

tests/py/dynamo/backend/test_backend_compiler.py

@@ -1,7 +1,8 @@
-from utils import lower_graph_testing
-from torch.testing._internal.common_utils import run_tests, TestCase
 import torch
 import torch_tensorrt
+from torch.testing._internal.common_utils import TestCase, run_tests
+
+from ..testing_utilities import lower_graph_testing
 
 
 class TestFakeTensors(TestCase):

tests/py/dynamo/backend/test_pre_aot_lowering.py

@@ -3,7 +3,7 @@
 from torch.testing._internal.common_utils import run_tests
 from torch_tensorrt import Input
 
-from harness import DispatchTestCase
+from .harness import DispatchTestCase
 
 
 class TestAdaptiveAvgPoolConverter(DispatchTestCase):

tests/py/dynamo/backend/test_specialized_models.py

@@ -1,8 +1,9 @@
 import torch
 from torch.testing._internal.common_utils import run_tests
-from harness import DispatchTestCase
 from torch_tensorrt import Input
 
+from .harness import DispatchTestCase
+
 
 class TestBatchNormConverter(DispatchTestCase):
     def test_batchnorm(self):

tests/py/dynamo/converters/__init__.py

@@ -1,14 +1,14 @@
-from typing import Callable
 import unittest
+from typing import Callable
 
 import torch
 import torch.nn as nn
-
 from parameterized import parameterized
 from torch.testing._internal.common_utils import run_tests
-from harness import DispatchTestCase
 from torch_tensorrt import Input
 
+from .harness import DispatchTestCase
+
 NEED_TEST_BOTH_CONSTANTS_CASE = True
 
 elementwise_ops = [