Skip to content
This repository was archived by the owner on Sep 18, 2025. It is now read-only.
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 1 addition & 8 deletions ci/run_benchmark_pytests.sh
Original file line number Diff line number Diff line change
Expand Up @@ -8,12 +8,5 @@ set -euo pipefail
pushd /
timeout 1m python -m ucp.benchmarks.send_recv -o cupy --server-dev 0 --client-dev 0 --reuse-alloc --backend ucp-async
timeout 1m python -m ucp.benchmarks.send_recv -o cupy --server-dev 0 --client-dev 0 --reuse-alloc --backend ucp-core

# TODO: run cudf_merge benchmark unconditionally once there are CUDA 13 cudf packages
# ref: https://github.com/rapidsai/ucx-py/pull/1162#issuecomment-3210841342
if [[ $(python -c "import cudf") ]]; then
timeout 1m python -m ucp.benchmarks.cudf_merge --chunks-per-dev 4 --chunk-size 10000 --rmm-init-pool-size 2097152
else
echo "'cudf' not installed, skipping cudf_merge benchmark"
fi
timeout 1m python -m ucp.benchmarks.cudf_merge --chunks-per-dev 4 --chunk-size 10000 --rmm-init-pool-size 2097152
popd
18 changes: 4 additions & 14 deletions dependencies.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -255,27 +255,16 @@ dependencies:
- matrix:
packages:
- numba-cuda[cu13]>=0.19.0,<0.20.0a0
# TODO: re-include cudf on CUDA 13 here once cudf CUDA 13 packages exist
# ref: https://github.com/rapidsai/ucx-py/pull/1162#issuecomment-3210841342
depends_on_cudf:
common:
- output_types: [conda]
packages:
- &cudf_unsuffixed cudf==25.10.*,>=0.0.0a0
- output_types: requirements
packages:
# pip recognizes the index as a global option for the requirements.txt file
- --extra-index-url=https://pypi.anaconda.org/rapidsai-wheels-nightly/simple
specific:
- output_types: [conda]
matrices:
- matrix:
cuda: "12.*"
packages:
- &cudf_unsuffixed cudf==25.10.*,>=0.0.0a0
- matrix:
cuda: "13.*"
packages:
- matrix:
packages:
- *cudf_unsuffixed
- output_types: [requirements, pyproject]
matrices:
- matrix:
Expand All @@ -286,4 +275,5 @@ dependencies:
- matrix:
cuda: "13.*"
packages:
- cudf-cu13==25.10.*,>=0.0.0a0
- {matrix: null, packages: [*cudf_unsuffixed]}