Migrate onnx_ptq bash tests to github as well

Signed-off-by: Keval Morabia <[email protected]>

Author: kevalmorabia97

.github/workflows/_example_tests_runner.yml

@@ -27,6 +27,11 @@ on:
         required: false
         type: string
         default: "linux-amd64-gpu-h100-latest-1"
+      test_type:
+        description: "Test type: 'pytest' or 'bash'"
+        required: false
+        type: string
+        default: "pytest"
 
 jobs:
   run-test:
@@ -60,5 +65,8 @@ jobs:
           find examples/${{ inputs.example }} -name "requirements.txt" | while read req_file; do pip install -r "$req_file" || exit 1; done
       - name: Run tests
         run: |
-          echo "Running tests for: ${{ inputs.example }}"
-          pytest tests/examples/${{ inputs.example }}
+          if [ "${{ inputs.test_type }}" = "bash" ]; then
+            bash tests/examples/test_${{ inputs.example }}.sh
+          else
+            pytest tests/examples/${{ inputs.example }}
+          fi

.github/workflows/example_tests.yml

@@ -140,6 +140,22 @@ jobs:
       pip_install_extras: "[all,dev-test]"
       runner: linux-amd64-gpu-l4-latest-1
 
+  # TODO: convert test to pytest as well so we can remove this job
+  example-onnx-bash-non-pr:
+    if: ${{ !startsWith(github.ref, 'refs/heads/pull-request/') }}
+    strategy:
+      matrix:
+        example: [onnx_ptq]
+    uses: ./.github/workflows/_example_tests_runner.yml
+    secrets: inherit
+    with:
+      docker_image: "nvcr.io/nvidia/tensorrt:25.08-py3"
+      example: ${{ matrix.example }}
+      timeout_minutes: 90
+      pip_install_extras: "[all,dev-test]"
+      runner: linux-amd64-gpu-l4-latest-1
+      test_type: bash
+
   ##### Required Check for PR #####
   example-pr-required-check:
     # Run even if example tests are skipped

.gitlab/tests.yml

@@ -1,36 +1,14 @@
-# NOTE: Make sure this file is consistent with .github/workflows/{unit,gpu,example}_tests.yml
-.tests-default:
-  variables:
-    PIP_CONSTRAINT: "" # Disable pip constraint for upgrading packages
+##### Megatron / NeMo Integration Tests #####
+megatron-nemo-integration:
+  extends: .tests-default
   stage: tests
+  variables:
+    UPSTREAM_REF: $CI_COMMIT_REF_NAME
   rules:
     - if: $CI_PIPELINE_SOURCE == "schedule"
     - if: $CI_COMMIT_TAG =~ /^\d+\.\d+\.\d+$/
     - if: $CI_PIPELINE_SOURCE == "web" || $CI_MERGE_REQUEST_TARGET_BRANCH_PROTECTED == "true"
       when: manual
-
-##### Example Tests #####
-example-onnx-bash:
-  extends: .tests-default
-  timeout: 90m
-  image: nvcr.io/nvidia/tensorrt:25.08-py3
-  tags: [docker, linux, 2-gpu, sm>=89]
-  parallel:
-    matrix:
-      - EXAMPLE: [onnx_ptq]
-  before_script:
-    # Add libcudnn*.so and libnv*.so to path
-    - export LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:/usr/include:/usr/lib/x86_64-linux-gnu"
-  script:
-    - pip install ".[all,dev-test]"
-    - find examples/$EXAMPLE -name "requirements.txt" | while read req_file; do pip install -r "$req_file" || exit 1; done
-    - bash tests/examples/test_$EXAMPLE.sh
-
-##### Megatron / NeMo Integration Tests #####
-megatron-nemo-integration:
-  extends: .tests-default
-  variables:
-    UPSTREAM_REF: $CI_COMMIT_REF_NAME
   trigger:
     project: omniml/integration/nmm-sandbox
     branch: main