diff --git a/.github/workflows/pr.yaml b/.github/workflows/pr.yaml index dd4482375b9..3d0cda92f5e 100644 --- a/.github/workflows/pr.yaml +++ b/.github/workflows/pr.yaml @@ -113,7 +113,7 @@ jobs: build_type: pull-request package-name: cudf # Install cupy-cuda11x for arm from a special index url - test-before-arm64: "python -m pip install cupy-cuda11x -f https://pip.cupy.dev/aarch64" + test-before-arm64: "python -m pip install 'cupy-cuda11x<12.0.0' -f https://pip.cupy.dev/aarch64" test-unittest: "python -m pytest -v -n 8 ./python/cudf/cudf/tests" test-smoketest: "python ./ci/wheel_smoke_test_cudf.py" wheel-build-dask-cudf: diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml index a4bd14439b0..660f46199b4 100644 --- a/.github/workflows/test.yaml +++ b/.github/workflows/test.yaml @@ -86,7 +86,7 @@ jobs: date: ${{ inputs.date }} sha: ${{ inputs.sha }} package-name: cudf - test-before-arm64: "python -m pip install cupy-cuda11x -f https://pip.cupy.dev/aarch64" + test-before-arm64: "python -m pip install 'cupy-cuda11x<12.0.0' -f https://pip.cupy.dev/aarch64" test-unittest: "python -m pytest -v -n 8 ./python/cudf/cudf/tests" wheel-tests-dask-cudf: secrets: inherit