Skip to content

Commit

Permalink
fix cugraph-ops
Browse files Browse the repository at this point in the history
  • Loading branch information
jameslamb committed Sep 19, 2024
1 parent 733d01b commit c28d198
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 8 deletions.
9 changes: 5 additions & 4 deletions ci/use_conda_packages_from_prs.sh
Original file line number Diff line number Diff line change
Expand Up @@ -11,8 +11,9 @@ UCXX_CHANNEL=$(rapids-get-pr-conda-artifact ucxx 278 cpp)
LIBRAFT_CHANNEL=$(rapids-get-pr-conda-artifact raft 2433 cpp)
RAFT_CHANNEL=$(rapids-get-pr-conda-artifact raft 2433 python)

LIBCUGRAPHOPS_CHANNEL=$(rapids-get-pr-conda-artifact cugraph-ops 692 cpp)
PYLIBCUGRAPHOPS_CHANNEL=$(rapids-get-pr-conda-artifact cugraph-ops 692 python)
# NOTE: rapids-get-pr-wheel-artifact seems to not work with private repos
# LIBCUGRAPHOPS_CHANNEL=$(rapids-get-pr-conda-artifact cugraph-ops 692 cpp)
# PYLIBCUGRAPHOPS_CHANNEL=$(rapids-get-pr-conda-artifact cugraph-ops 692 python)

conda config --system --add channels "${LIBRMM_CHANNEL}"
conda config --system --add channels "${RMM_CHANNEL}"
Expand All @@ -21,5 +22,5 @@ conda config --system --add channels "${CUDF_PYTHON_CHANNEL}"
conda config --system --add channels "${UCXX_CHANNEL}"
conda config --system --add channels "${LIBRAFT_CHANNEL}"
conda config --system --add channels "${RAFT_CHANNEL}"
conda config --system --add channels "${LIBCUGRAPHOPS_CHANNEL}"
conda config --system --add channels "${PYLIBCUGRAPHOPS_CHANNEL}"
# conda config --system --add channels "${LIBCUGRAPHOPS_CHANNEL}"
# conda config --system --add channels "${PYLIBCUGRAPHOPS_CHANNEL}"
10 changes: 6 additions & 4 deletions ci/use_wheels_from_prs.sh
Original file line number Diff line number Diff line change
Expand Up @@ -41,9 +41,10 @@ PYLIBRAFT_CHANNEL=$(
RAPIDS_PY_WHEEL_NAME=pylibraft_${RAPIDS_PY_CUDA_SUFFIX} rapids-get-pr-wheel-artifact raft 2433 python
)

PYLIBCUGRAPHOPS_CHANNEL=$(
RAPIDS_PY_WHEEL_NAME=pylibraft_${RAPIDS_PY_CUDA_SUFFIX} rapids-get-pr-wheel-artifact cugraph-ops 692 python
)
# NOTE: rapids-get-pr-wheel-artifact seems to not work with private repos
# PYLIBCUGRAPHOPS_CHANNEL=$(
# RAPIDS_PY_WHEEL_NAME=pylibraft_${RAPIDS_PY_CUDA_SUFFIX} rapids-get-pr-wheel-artifact cugraph-ops 692 python
# )

cat > /tmp/constraints.txt <<EOF
librmm-${RAPIDS_PY_CUDA_SUFFIX} @ file://$(echo ${LIBRMM_CHANNEL}/librmm_*.whl)
Expand All @@ -57,7 +58,8 @@ libucxx-${RAPIDS_PY_CUDA_SUFFIX} @ file://$(echo ${LIBUCXX_CHANNEL}/libucxx_*.wh
distributed-ucxx-${RAPIDS_PY_CUDA_SUFFIX} @ file://$(echo ${DISTRIBUTED_UCXX_CHANNEL}/distributed_ucxx_*.whl)
raft-dask-${RAPIDS_PY_CUDA_SUFFIX} @ file://$(echo ${RAFT_DASK_CHANNEL}/raft_dask_*.whl)
pylibraft-${RAPIDS_PY_CUDA_SUFFIX} @ file://$(echo ${PYLIBRAFT_CHANNEL}/pylibraft_*.whl)
pylibcugraphops-${RAPIDS_PY_CUDA_SUFFIX} @ file://$(echo ${PYLIBCUGRAPHOPS_CHANNEL}/pylibcugraphops_*.whl)
EOF

# pylibcugraphops-${RAPIDS_PY_CUDA_SUFFIX} @ file://$(echo ${PYLIBCUGRAPHOPS_CHANNEL}/pylibcugraphops_*.whl)

export PIP_CONSTRAINT=/tmp/constraints.txt

0 comments on commit c28d198

Please sign in to comment.