fix gpt_test.yml

Signed-off-by: Kinjal Patel <kinjalpravin@nvidia.com>

Author: kinjalpatel27

.github/workflows/gpu_tests.yml

@@ -61,12 +61,26 @@ jobs:
         HF_TOKEN: ${{ secrets.HF_TOKEN }}
     steps:
       - uses: actions/checkout@v6
-      - uses: ./.github/actions/gpu-test-run
+      - uses: nv-gha-runners/setup-proxy-cache@main
+      - name: Setup environment variables
+        run: |
+          echo "LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/include:/usr/lib/x86_64-linux-gnu" >> $GITHUB_ENV
+      - name: Run gpu tests
+        env:
+          COVERAGE_PROCESS_START: ${{ github.workspace }}/pyproject.toml
+          COVERAGE_FILE: ${{ github.workspace }}/.coverage
+        run: |
+          python3 -m pip install nox && nox -s ${{ matrix.example }}
+      - name: Upload GPU coverage to Codecov
+        uses: codecov/codecov-action@v5
         with:
-          example: ${{ matrix.example }}
-          codecov_token: ${{ secrets.CODECOV_TOKEN }}
+          token: ${{ secrets.CODECOV_TOKEN }}
+          files: coverage.xml
+          flags: gpu
+          fail_ci_if_error: false # test may be skipped if relevant file changes are not detected
+          verbose: true
 
-  # Docker Hub image: anonymous pull (no ``credentials:``) and no coverage
+  # Docker Hub image: anonymous pull and no coverage
   gpu-tests-vllm:
     needs: [pr-gate]
     if: needs.pr-gate.outputs.any_changed == 'true'
@@ -81,10 +95,13 @@ jobs:
         HF_TOKEN: ${{ secrets.HF_TOKEN }}
     steps:
       - uses: actions/checkout@v6
-      - uses: ./.github/actions/gpu-test-run
-        with:
-          example: gpu_vllm
-          with_coverage: "false"
+      - uses: nv-gha-runners/setup-proxy-cache@main
+      - name: Setup environment variables
+        run: |
+          echo "LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/include:/usr/lib/x86_64-linux-gnu" >> $GITHUB_ENV
+      - name: Run gpu tests
+        run: |
+          python3 -m pip install nox && nox -s gpu_vllm
 
   gpu-pr-required-check:
     # Run even if any of the gpu jobs is skipped