Skip to content

Commit

Permalink
fix rmm pinning, add make, standardize quotes
Browse files Browse the repository at this point in the history
  • Loading branch information
gforsyth committed Feb 13, 2025
1 parent bae1026 commit e9d4f16
Showing 1 changed file with 19 additions and 17 deletions.
36 changes: 19 additions & 17 deletions conda/recipes/ucxx/recipe.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -48,12 +48,12 @@ cache:
CMAKE_CXX_COMPILER_LAUNCHER: ${{ env.get("CMAKE_CXX_COMPILER_LAUNCHER") }}
CMAKE_GENERATOR: ${{ env.get("CMAKE_GENERATOR") }}
PARALLEL_LEVEL: ${{ env.get("PARALLEL_LEVEL") }}
SCCACHE_BUCKET: ${{ env.get("SCCACHE_BUCKET", default="") }}
SCCACHE_IDLE_TIMEOUT: ${{ env.get("SCCACHE_IDLE_TIMEOUT", default="") }}
SCCACHE_REGION: ${{ env.get("SCCACHE_REGION", default="") }}
SCCACHE_S3_USE_SSL: ${{ env.get("SCCACHE_S3_USE_SSL", default="") }}
SCCACHE_S3_NO_CREDENTIALS: ${{ env.get("SCCACHE_S3_NO_CREDENTIALS", default="") }}
SCCACHE_S3_KEY_PREFIX: libucxx-${{ env.get("RAPIDS_CONDA_ARCH", default="") }}
SCCACHE_BUCKET: ${{ env.get("SCCACHE_BUCKET") }}
SCCACHE_IDLE_TIMEOUT: ${{ env.get("SCCACHE_IDLE_TIMEOUT") }}
SCCACHE_REGION: ${{ env.get("SCCACHE_REGION") }}
SCCACHE_S3_USE_SSL: ${{ env.get("SCCACHE_S3_USE_SSL") }}
SCCACHE_S3_NO_CREDENTIALS: ${{ env.get("SCCACHE_S3_NO_CREDENTIALS") }}
SCCACHE_S3_KEY_PREFIX: libucxx-${{ env.get("RAPIDS_CONDA_ARCH") }}

outputs:
- package:
Expand All @@ -70,7 +70,7 @@ outputs:
run:
- if: cuda_major == "11"
then: cudatoolkit
- ${{ pin_compatible('cuda-version', upper_bound='x', lower_bound='x') }}
- ${{ pin_compatible("cuda-version", upper_bound="x", lower_bound="x") }}
- ucx >=1.15.0,<1.19.0
tests:
- package_contents:
Expand Down Expand Up @@ -118,7 +118,7 @@ outputs:
host:
- cmake ${{ cmake_version }}
run:
- ${{ pin_subpackage('libucxx', exact=True) }}
- ${{ pin_subpackage("libucxx", exact=True) }}
about:
homepage: ${{ load_from_file("python/libucxx/pyproject.toml").project.urls.Homepage }}
license: ${{ load_from_file("python/libucxx/pyproject.toml").project.license.text }}
Expand All @@ -134,16 +134,16 @@ outputs:
- cuda-version =${{ cuda_version }}
- cmake ${{ cmake_version }}
run:
- ${{ pin_subpackage('libucxx', exact=True) }}
- ${{ pin_subpackage("libucxx", exact=True) }}
- if: cuda_major == "11"
then:
- cudatoolkit
else:
- cuda-cudart
- ${{ pin_compatible('cuda-version', upper_bound='x', lower_bound='x') }}
- ${{ pin_compatible("cuda-version", upper_bound="x", lower_bound="x") }}
ignore_run_exports:
from_package:
- ${{ compiler('cuda') }}
- ${{ compiler("cuda") }}
about:
homepage: ${{ load_from_file("python/libucxx/pyproject.toml").project.urls.Homepage }}
license: ${{ load_from_file("python/libucxx/pyproject.toml").project.license.text }}
Expand All @@ -162,31 +162,33 @@ outputs:
- pip
- scikit-build-core >=0.10.0
- cython >=3.0.0
- ${{ pin_subpackage('libucxx', exact=True) }}
- ${{ pin_subpackage("libucxx", exact=True) }}
- ucx
- rapids-build-backend >=0.3.0,<0.4.0.dev0
- rmm ${{ rapids_version }}
- cuda-version =${{ cuda_version }}
- cmake ${{ cmake_version }}
# Without this we get an error that there is no build program corresponding to "Unix Makefiles"
- make
run:
- python * *_cpython
- ${{ pin_compatible('cuda-version', upper_bound='x', lower_bound='x') }}
- ${{ pin_compatible("cuda-version", upper_bound="x", lower_bound="x") }}
- if: cuda_major == "11"
then:
- cudatoolkit
else:
- cuda-cudart
- ucx >=1.15.0,<1.19.0
- ${{ pin_subpackage('libucxx', exact=True) }}
- ${{ pin_subpackage('rmm', upper_bound="x.x") }}
- ${{ pin_subpackage("libucxx", exact=True) }}
- ${{ pin_compatible("rmm", upper_bound="x.x") }}
- numpy >=1.23,<3.0a0
- pynvml >=12.0.0,<13.0.0a0
run_constraints:
- cupy >=9.5.0
- numba >=0.59.1,<0.61.0a0
ignore_run_exports:
from_package:
- ${{ compiler('cuda') }}
- ${{ compiler("cuda") }}
- ucx
tests:
- script:
Expand Down Expand Up @@ -222,7 +224,7 @@ outputs:
run:
- python * *_cpython
- rapids-dask-dependency ${{ rapids_version }}
- ${{ pin_subpackage('ucxx', upper_bound='x.x') }}
- ${{ pin_subpackage("ucxx", upper_bound="x.x") }}
tests:
- python:
imports:
Expand Down

0 comments on commit e9d4f16

Please sign in to comment.