ci: Update QA L1 PYTHORCH Unittest

Author: yefubao

.github/workflows/qa_l1_pytorch_unittest.yml

@@ -74,12 +74,12 @@ jobs:
           echo "===== Available memory ====="
           free -h
 
-      # - name: Install dependencies and build transformer_engine
-      #   timeout-minutes: 30
-      #   run: |
-      #     echo "\n=============== Install transformer_engine ==============="
-      #     pip install --no-build-isolation -vvv . --no-deps
-      #     cp -rf transformer_engine/plugins/ /usr/local/lib/python3.10/dist-packages/transformer_engine/plugins/
+      - name: Install dependencies and build transformer_engine
+        timeout-minutes: 30
+        run: |
+          echo "\n=============== Install transformer_engine ==============="
+          pip install --no-build-isolation -vvv . --no-deps
+          cp -rf transformer_engine/plugins/ /usr/local/lib/python3.10/dist-packages/transformer_engine/plugins/
 
       - name: Verify installation
         shell: bash
@@ -117,7 +117,31 @@ jobs:
         env:
           XML_LOG_DIR: "/logs/pytorch/onnx"
           TE_PATH: .
-        run: bash ./qa/L1_pytorch_onnx_unittest/test.sh
+        run: |
+          SKIP_TESTS=( 
+            "test_trt_integration[None]"
+          )
+
+          TARGET_FILE="./qa/L1_pytorch_onnx_unittest/test.sh"
+          OLD_CMD="python3 -m pytest --tb=auto --junitxml=\$XML_LOG_DIR/test_onnx_export.xml \$TE_PATH/tests/pytorch/test_onnx_export.py"
+
+          SKIP_PARAM=""
+          if [ ${#SKIP_TESTS[@]} -gt 0 ]; then
+            SKIP_EXPR=""
+            for test in "${SKIP_TESTS[@]}"; do
+              if [ -z "$SKIP_EXPR" ]; then
+                SKIP_EXPR="not $test"
+              else
+                SKIP_EXPR+=" and not $test"
+              fi
+            done
+            SKIP_PARAM="-k \"${SKIP_EXPR}\""
+          fi
+
+          CMD="${OLD_CMD} ${SKIP_PARAM}"
+          sed -i "s|${OLD_CMD}|${CMD}|g" "${TARGET_FILE}"
+
+          bash ./qa/L1_pytorch_onnx_unittest/test.sh
 
       - name: Upload Installation Logs
         if: always() && (steps.L1_pytorch_distributed_unittest.outcome == 'failure' || steps.L1_pytorch_onnx_unittest.outcome == 'failure')