skip Cython tests when using CUDA wheels

NVIDIA · Jan 9, 2025 · d57f5fc · d57f5fc
1 parent f38e1eb
commit d57f5fc
Showing 1 changed file with 13 additions and 8 deletions.
diff --git a/.github/workflows/test-wheel.yml b/.github/workflows/test-wheel.yml
@@ -163,14 +163,19 @@ jobs:
           pushd ./cuda_bindings
           pip install -r requirements.txt
           pytest -rxXs tests/
-          if [[ "${{ inputs.host-platform }}" == linux* ]]; then
-            bash tests/cython/build_tests.sh
-          elif [[ "${{ inputs.host-platform }}" == win* ]]; then
-            # TODO: enable this once win-64 runners are up 
-            exit 1
-          fi  
-          pytest -rxXs tests/cython
-          popd
+
+          # It is a bit convoluted to run the Cython tests against CTK wheels,
+          # so let's just skip them.
+          if [[ "${{ inputs.local-ctk }}" == 1 ]]; then
+            if [[ "${{ inputs.host-platform }}" == linux* ]]; then
+              bash tests/cython/build_tests.sh
+            elif [[ "${{ inputs.host-platform }}" == win* ]]; then
+              # TODO: enable this once win-64 runners are up 
+              exit 1
+            fi  
+            pytest -rxXs tests/cython
+            popd
+          fi
 
       - name: Run cuda.core tests
         run: |