chore: Modify dynamo fx_ts_compat tests

Signed-off-by: Dheeraj Peri <[email protected]>

Author: peri044

.circleci/config.yml

@@ -727,184 +727,14 @@ commands:
       - store_artifacts:
           path: /tmp/testlogs
 
-  test-dynamo-fx_ts_converters_acc:
-    description: "Test the Dynamo acc converters"
-    steps:
-      - run:
-          name: Run FX converter tests
-          command: |
-            cd py/torch_tensorrt/dynamo/fx_ts_compat/test
-            pushd converters/acc_op/
-            pytest --junitxml=/tmp/artifacts/test_results/dynamo/converters/acc_op/test_results.xml
-            popd
-
-      - store_test_results:
-          path: /tmp/artifacts
-      - store_artifacts:
-          path: /tmp/testlogs
-
-  test-dynamo-fx_ts_converters_aten:
-    description: "Test the dynamo aten converters"
-    steps:
-      - run:
-          name: Run dynamo converter tests
-          command: |
-            cd py/torch_tensorrt/dynamo/fx_ts_compat/test
-            pushd converters/aten_op/
-            pytest --junitxml=/tmp/artifacts/test_results/dynamo/converters/aten_op/test_results.xml
-            popd
-
-      - store_test_results:
-          path: /tmp/artifacts
-      - store_artifacts:
-          path: /tmp/testlogs
-
-  test-dynamo-fx_ts_converters_vanilla:
-    description: "Test the dynamo vanilla converters"
-    steps:
-      - run:
-          name: Run dynamo converter tests
-          command: |
-            cd py/torch_tensorrt/dynamo/fx_ts_compat/test
-            pushd converters/vanilla/
-            pytest --junitxml=/tmp/artifacts/test_results/dynamo/converters/vanilla/test_results.xml
-            popd
-
-      - store_test_results:
-          path: /tmp/artifacts
-      - store_artifacts:
-          path: /tmp/testlogs
-
-  test-dynamo-fx_ts_passes:
-    description: "Test the dynamo passes"
-    steps:
-      - run:
-          name: Run dynamo passes
-          command: |
-            cd py/torch_tensorrt/dynamo/fx_ts_compat/test
-            pushd passes
-            list_passes=$(ls | grep -v test_setitem*)
-            pytest $list_passes --junitxml=/tmp/artifacts/test_results/dynamo/passes/test_results.xml
-            popd
-      - store_test_results:
-          path: /tmp/artifacts
-      - store_artifacts:
-          path: /tmp/testlogs
-
-  test-dynamo-fx_ts_tools:
-    description: "Test the dynamo tools"
-    steps:
-      - run:
-          name: Run dynamo tools
-          command: |
-            cd py/torch_tensorrt/dynamo/fx_ts_compat/test
-            pushd tools
-            pytest --junitxml=/tmp/artifacts/test_results/dynamo/tools/test_results.xml
-            popd
-      - store_test_results:
-          path: /tmp/artifacts
-      - store_artifacts:
-          path: /tmp/testlogs
-
-  test-dynamo-fx_ts_trt_lower:
-    description: "Test the dynamo TRT lowering"
-    steps:
-      - run:
-          name: Run dynamo TRT lowering
-          command: |
-            cd py/torch_tensorrt/dynamo/fx_ts_compat/test
-            pushd trt_lower
-            pytest --junitxml=/tmp/artifacts/test_results/dynamo/trt_lower/test_results.xml
-            popd
-      - store_test_results:
-          path: /tmp/artifacts
-      - store_artifacts:
-          path: /tmp/testlogs
-
-  test-dynamo-fx_ts_tracer:
-    description: "Test all dynamo tracers"
-    steps:
-      - run:
-          name: Run dynamo tracer
-          command: |
-            cd py/torch_tensorrt/dynamo/fx_ts_compat/test
-            pushd tracer
-            list_tracer=$(ls | grep -v test_dispatch_*)
-            pytest $list_tracer --junitxml=/tmp/artifacts/test_results/fx/tracer/test_results.xml
-            popd
-      - store_test_results:
-          path: /tmp/artifacts
-      - store_artifacts:
-          path: /tmp/testlogs
-
-  test-dynamo-fx_ts_tracer_acc:
-    description: "Test the dynamo acc tracer only"
-    steps:
-      - run:
-          name: Run dynamo tracer
-          command: |
-            cd py/torch_tensorrt/dynamo/fx_ts_compat/test
-            pushd tracer
-            list_tracer=$(ls | grep test_acc)
-            pytest $list_tracer --junitxml=/tmp/artifacts/test_results/dynamo/tracer/test_results.xml
-            popd
-      - store_test_results:
-          path: /tmp/artifacts
-      - store_artifacts:
-          path: /tmp/testlogs
-
-  test-dynamo-fx_ts_quant:
-    description: "Test the dynamo quant"
-    steps:
-      - run:
-          name: Run dynamo quant tests
-          command: |
-            cd py/torch_tensorrt/dynamo/fx_ts_compat/test
-            pushd quant/
-            pytest --junitxml=/tmp/artifacts/test_results/dynamo/quant/test_results.xml
-            popd
-
-      - store_test_results:
-          path: /tmp/artifacts
-      - store_artifacts:
-          path: /tmp/testlogs
-
   test-dynamo-fx_ts:
     description: "Test the dynamo backend"
     steps:
       - run:
           name: Run dynamo tests
           command: |
             mkdir -p /tmp/artifacts/test_results
-      - test-dynamo-fx_ts_converters_acc
-      - test-dynamo-fx_ts_converters_aten
-      - test-dynamo-fx_ts_converters_vanilla
-      - test-dynamo-fx_ts_passes
-      - test-dynamo-fx_ts_tools
-      - test-dynamo-fx_ts_trt_lower
-      - test-dynamo-fx_ts_tracer
-      - test-dynamo-fx_ts_core
-      - test-dynamo-fx_ts_quant
-      - store_test_results:
-          path: /tmp/artifacts
-      - store_artifacts:
-          path: /tmp/testlogs
-
-  test-dynamo-fx_ts-no-aten:
-    description: "Test the dynamo backend without aten operators"
-    steps:
-      - run:
-          name: Run dynamo tests without aten ops
-          command: |
-            mkdir -p /tmp/artifacts/test_results
-      - test-dynamo-fx_ts_converters_acc
-      - test-dynamo-fx_ts_converters_vanilla
-      - test-dynamo-fx_ts_passes
-      - test-dynamo-fx_ts_tools
-      - test-dynamo-fx_ts_trt_lower
-      - test-dynamo-fx_ts_tracer_acc
       - test-dynamo-fx_ts_core
-      - test-dynamo-fx_ts_quant
       - store_test_results:
           path: /tmp/artifacts
       - store_artifacts:
@@ -1119,37 +949,6 @@ jobs:
       - dump-test-env
       - test-dynamo-fx_ts
 
-  test-py-dynamo-x86_64-linux-no-aten:
-    parameters:
-      torch-build:
-        type: string
-      torch-build-index:
-        type: string
-      trt-version-long:
-        type: string
-    machine:
-      image: ubuntu-2004-cuda-11.4:202110-01
-    resource_class: gpu.nvidia.large
-    steps:
-      - checkout
-      - attach_workspace:
-          at: /tmp/dist/
-      - install-torch-from-index:
-          torch-build:  << parameters.torch-build >>
-          torch-build-index: << parameters.torch-build-index >>
-      - create-py-env:
-          trt-version-long: << parameters.trt-version-long >>
-      - install-cudnn
-      # - run:
-      #     name: "Set LD_LIBRARY_PATH path to include the installed CUDNN"
-      #     command: export LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu/:$LD_LIBRARY_PATH
-      - run:
-          name: "Install torch-tensorrt"
-          command: pip3 install --pre /tmp/dist/x86_64-linux/*cp39-cp39*.whl
-      # We install torch after torch-trt because pip automatically enforces the version constraint otherwise
-      - dump-test-env
-      - test-dynamo-fx_ts-no-aten
-
   package-x86_64-linux:
     parameters:
       enabled:
@@ -1565,12 +1364,6 @@ workflows:
           requires:
             - build-x86_64-linux-legacy
 
-      - test-py-dynamo-x86_64-linux-no-aten:
-          torch-build: << pipeline.parameters.torch-build-legacy >>
-          torch-build-index: << pipeline.parameters.torch-build-index-legacy >>
-          trt-version-long: << pipeline.parameters.trt-version-long >>
-          requires:
-            - build-x86_64-linux-legacy
   release:
     when: << pipeline.parameters.enable-packaging >>
     jobs: