install cuda-python for doc build

NVIDIA · Jan 13, 2025 · 94ba0b7 · 94ba0b7
1 parent ef09568
commit 94ba0b7
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/.github/workflows/build-docs.yml b/.github/workflows/build-docs.yml
@@ -70,6 +70,17 @@ jobs:
           echo "CUDA_BINDINGS_ARTIFACT_NAME=${CUDA_BINDINGS_ARTIFACT_BASENAME}-${{ github.sha }}" >> $GITHUB_ENV
           echo "CUDA_BINDINGS_ARTIFACTS_DIR=$(realpath "$REPO_DIR/cuda_bindings/dist")" >> $GITHUB_ENV
 
+      - name: Download cuda-python build artifacts
+        uses: actions/download-artifact@v4
+        with:
+          name: cuda-python-wheel
+          path: .
+
+      - name: Display structure of downloaded cuda-python artifacts
+        run: |
+          pwd
+          ls -lahR .
+
       - name: Download cuda.bindings build artifacts
         uses: actions/download-artifact@v4
         with:
@@ -102,6 +113,8 @@ jobs:
           pip install *.whl
           popd
 
+          pip install cuda_python*.whl
+
       - name: Build all (latest) docs
         id: build
         run: |