Upload wheel to artifact storage

NVIDIA · Jul 23, 2024 · a4af89b · a4af89b
1 parent 877676c
commit a4af89b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/.github/workflows/wheels-build.yaml b/.github/workflows/wheels-build.yaml
@@ -158,5 +158,9 @@ jobs:
         # Use a shell that loads the rc file so that we get the compiler settings
         shell: bash -leo pipefail {0}
       - name: Upload additional artifacts
+        id: upload-wheel
         if: "!cancelled()"
-        run: rapids-upload-artifacts-dir cuda${RAPIDS_CUDA_VERSION%%.*}_$(arch)_py${RAPIDS_PY_VERSION//.}
+        uses: actions/upload-artifact@v4
+        with:
+          name ${{ env.package_name }}
+          path: ${{ env.package_path }}/${{ env.package_name }}
diff --git a/ci/build_wheel.sh b/ci/build_wheel.sh
@@ -7,5 +7,5 @@ rapids-logger "Build wheel"
 mkdir -p ./dist
 python -m pip wheel . --wheel-dir=./dist -vvv --disable-pip-version-check --no-deps
 
-rapids-logger "Upload Wheel"
-RAPIDS_PY_WHEEL_NAME="numba_cuda" rapids-upload-wheels-to-s3 ./dist
+echo "package_name=$(basename numba_cuda-*.whl)" >> $GITHUB_ENV
+echo "package_path=`pwd`" >> $GITHUB_ENV