Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[REVIEW] Upgrade dask & distributed minimum version #4632

Merged
merged 12 commits into from
Mar 17, 2022
Merged

[REVIEW] Upgrade dask & distributed minimum version #4632

Show file tree
Hide file tree
Changes from 7 commits
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
fc1eb1a
upgrade dask
galipremsagar Mar 14, 2022
77536a2
Merge branch 'rapidsai:branch-22.04' into upgrade_dask
galipremsagar Mar 15, 2022
3c2dc23
Merge branch 'rapidsai:branch-22.04' into upgrade_dask
galipremsagar Mar 15, 2022
4a84761
temp commit
galipremsagar Mar 16, 2022
07ef5f7
Adding testing label for pyraft
cjnolet Mar 16, 2022
c9b64a6
Update build.sh
galipremsagar Mar 16, 2022
1e6add5
Update build.sh
galipremsagar Mar 16, 2022
e9eeaf7
Update ci/cpu/build.sh
galipremsagar Mar 16, 2022
071b2cd
Update ci/cpu/build.sh
galipremsagar Mar 16, 2022
f8d5005
Update ci/cpu/build.sh
galipremsagar Mar 16, 2022
e717fbf
Apply suggestions from code review
galipremsagar Mar 16, 2022
c66723e
Apply suggestions from code review
galipremsagar Mar 16, 2022
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: 7 additions & 2 deletions ci/cpu/build.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -103,10 +103,15 @@ sccache --show-stats
if [ "$BUILD_CUML" == '1' ]; then
if [[ -z "$PROJECT_FLASH" || "$PROJECT_FLASH" == "0" ]]; then
gpuci_logger "Build conda pkg for cuml"
gpuci_conda_retry build --croot ${CONDA_BLD_DIR} conda/recipes/cuml --python=${PYTHON}
gpuci_logger "TEMP COMMIT: using mamba build"
conda create -n tmp -c gpuci -c conda-forge gpuci-tools boa
gpuci_conda_retry mambabuild --croot ${CONDA_BLD_DIR} conda/recipes/cuml --python=${PYTHON}
else
gpuci_logger "PROJECT FLASH: Build conda pkg for cuml"
gpuci_conda_retry build --no-build-id --croot ${CONDA_BLD_DIR} conda/recipes/cuml -c ${CONDA_LOCAL_CHANNEL} --dirty --no-remove-work-dir --python=${PYTHON}

gpuci_logger "FLASH TEMP COMMIT: using mamba build"
conda create -n tmp -c gpuci -c conda-forge gpuci-tools boa
gpuci_conda_retry mambabuild --no-build-id --croot ${CONDA_BLD_DIR} conda/recipes/cuml -c ${CONDA_LOCAL_CHANNEL} --dirty --no-remove-work-dir --python=${PYTHON}
mkdir -p ${CONDA_BLD_DIR}/cuml
mv ${CONDA_BLD_DIR}/work/ ${CONDA_BLD_DIR}/cuml/work
fi
Expand Down
2 changes: 1 addition & 1 deletion ci/gpu/build.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -53,7 +53,7 @@ gpuci_logger "Activate conda env"
conda activate rapids

gpuci_logger "Install dependencies"
gpuci_mamba_retry install -c conda-forge -c rapidsai -c rapidsai-nightly -c nvidia \
gpuci_mamba_retry install -c conda-forge -c rapidsai -c rapidsai-nightly -c nvidia -c rapidsai-nightly/label/testing \
"cudatoolkit=${CUDA_REL}" \
"cudf=${MINOR_VERSION}" \
"rmm=${MINOR_VERSION}" \
Expand Down
4 changes: 2 additions & 2 deletions conda/recipes/cuml/meta.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -51,8 +51,8 @@ requirements:
- nccl>=2.9.9
- ucx-py {{ ucx_py_version }}
- ucx-proc=*=gpu
- dask>=2021.11.1
- distributed>=2021.11.1
- dask>=2022.02.1
- distributed>=2022.02.1
- joblib >=0.11
- {{ pin_compatible('cudatoolkit', max_pin='x', min_pin='x') }}
- cuda-python >=11.5,<12.0
Expand Down