Skip to content

Commit

Permalink
Update conda packaging for CUDA 12.x.
Browse files Browse the repository at this point in the history
  • Loading branch information
@bdice
bdice committed Mar 6, 2024
1 parent b1c6a38 commit 808ff83
Show file tree
Hide file tree
Showing 2 changed files with 66 additions and 11 deletions.
6 changes: 6 additions & 0 deletions conda/recipes/cuvs/meta.yaml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,9 @@ build:
ignore_run_exports_from:
{% if cuda_major == "11" %}
- {{ compiler('cuda11') }}
{% else %}
- {{ compiler('cuda') }}
- cuda-cudart-dev
{% endif %}

requirements:
Expand All @@ -43,6 +46,7 @@ requirements:
- cudatoolkit
{% else %}
- cuda-python >=12.0,<13.0a0
- cuda-cudart-dev
{% endif %}
- cuda-version ={{ cuda_version }}
- cython >=3.0.0
Expand All @@ -57,6 +61,8 @@ requirements:
- {{ pin_compatible('cuda-version', max_pin='x', min_pin='x') }}
{% if cuda_major == "11" %}
- cudatoolkit
{% else %}
- cuda-cudart
{% endif %}
- pylibraft {{ minor_version }}
- libcuvs {{ version }}
Expand Down
71 changes: 60 additions & 11 deletions conda/recipes/libcuvs/meta.yaml
View file
Original file line number Diff line number Diff line change
@@ -1,13 +1,12 @@
# Copyright (c) 2022-2024, NVIDIA CORPORATION.

# Usage:
# conda build . -c conda-forge -c nvidia -c rapidsai
{% set version = environ['RAPIDS_PACKAGE_VERSION'].lstrip('v') + environ.get('VERSION_SUFFIX', '') %}
{% set minor_version = version.split('.')[0] + '.' + version.split('.')[1] %}
{% set cuda_version = '.'.join(environ['RAPIDS_CUDA_VERSION'].split('.')[:2]) %}
{% set cuda_major = cuda_version.split('.')[0] %}
{% set cuda_spec = ">=" + cuda_major ~ ",<" + (cuda_major | int + 1) ~ ".0a0" %} # i.e. >=11,<12.0a0
{% set date_string = environ['RAPIDS_DATE_STRING'] %}
# conda build . -c conda-forge -c nvidia -c rapidsai
{% set version = environ['RAPIDS_PACKAGE_VERSION'].lstrip('v') + environ.get('VERSION_SUFFIX', '') %}
{% set minor_version = version.split('.')[0] + '.' + version.split('.')[1] %}
{% set cuda_version = '.'.join(environ['RAPIDS_CUDA_VERSION'].split('.')[:2]) %}
{% set cuda_major = cuda_version.split('.')[0] %}
{% set date_string = environ['RAPIDS_DATE_STRING'] %}

package:
name: libcuvs-split
Expand Down Expand Up @@ -42,6 +41,13 @@ outputs:
ignore_run_exports_from:
{% if cuda_major == "11" %}
- {{ compiler('cuda11') }}
{% else %}
- {{ compiler('cuda') }}
- cuda-cudart-dev
- libcublas-dev
- libcurand-dev
- libcusolver-dev
- libcusparse-dev
{% endif %}
requirements:
build:
Expand Down Expand Up @@ -70,6 +76,7 @@ outputs:
- libcusparse {{ cuda11_libcusparse_host_version }}
- libcusparse-dev {{ cuda11_libcusparse_host_version }}
{% else %}
- cuda-cudart-dev
- cuda-profiler-api
- libcublas-dev
- libcurand-dev
Expand All @@ -79,6 +86,13 @@ outputs:
run:
- libraft ={{ minor_version }}
- {{ pin_compatible('cuda-version', max_pin='x', min_pin='x') }}
{% if cuda_major != "11" %}
- cuda-cudart
- libcublas
- libcurand
- libcusolver
- libcusparse
{% endif %}
about:
home: https://rapids.ai/
license: Apache-2.0
Expand All @@ -93,6 +107,13 @@ outputs:
ignore_run_exports_from:
{% if cuda_major == "11" %}
- {{ compiler('cuda11') }}
{% else %}
- {{ compiler('cuda') }}
- cuda-cudart-dev
- libcublas-dev
- libcurand-dev
- libcusolver-dev
- libcusparse-dev
{% endif %}
requirements:
build:
Expand Down Expand Up @@ -121,6 +142,7 @@ outputs:
- libcusparse {{ cuda11_libcusparse_host_version }}
- libcusparse-dev {{ cuda11_libcusparse_host_version }}
{% else %}
- cuda-cudart-dev
- cuda-profiler-api
- libcublas-dev
- libcurand-dev
Expand All @@ -130,6 +152,13 @@ outputs:
run:
- libraft ={{ minor_version }}
- {{ pin_compatible('cuda-version', max_pin='x', min_pin='x') }}
{% if cuda_major != "11" %}
- cuda-cudart
- libcublas
- libcurand
- libcusolver
- libcusparse
{% endif %}
about:
home: https://rapids.ai/
license: Apache-2.0
Expand All @@ -144,6 +173,13 @@ outputs:
ignore_run_exports_from:
{% if cuda_major == "11" %}
- {{ compiler('cuda11') }}
{% else %}
- {{ compiler('cuda') }}
- cuda-cudart-dev
- libcublas-dev
- libcurand-dev
- libcusolver-dev
- libcusparse-dev
{% endif %}
requirements:
build:
Expand All @@ -160,7 +196,7 @@ outputs:
- sysroot_{{ target_platform }} {{ sysroot_version }}
host:
- libraft ={{ minor_version }}
- libcuvs ={{ minor_version }}
- {{ pin_subpackage('libcuvs', exact=True) }}
- cuda-version ={{ cuda_version }}
{% if cuda_major == "11" %}
- cuda-profiler-api {{ cuda11_cuda_profiler_api_run_version }}
Expand All @@ -186,9 +222,15 @@ outputs:
- {{ pin_compatible('cuda-version', max_pin='x', min_pin='x') }}
{% if cuda_major == "11" %}
- cudatoolkit
{% else %}
- cuda-cudart
- libcublas
- libcurand
- libcusolver
- libcusparse
{% endif %}
- libraft ={{ minor_version }}
- libcuvs ={{ minor_version }}
- {{ pin_subpackage('libcuvs', exact=True) }}
- gmock {{ gtest_version }}
- gtest {{ gtest_version }}
about:
Expand All @@ -205,6 +247,10 @@ outputs:
ignore_run_exports_from:
{% if cuda_major == "11" %}
- {{ compiler('cuda11') }}
{% else %}
- {{ compiler('cuda') }}
- cuda-cudart-dev
- libcublas-dev
{% endif %}
requirements:
build:
Expand All @@ -221,7 +267,7 @@ outputs:
- sysroot_{{ target_platform }} {{ sysroot_version }}
host:
- libraft ={{ minor_version }}
- libcuvs ={{ minor_version }}
- {{ pin_subpackage('libcuvs', exact=True) }}
- cuda-version ={{ cuda_version }}
{% if cuda_major == "11" %}
- cuda-profiler-api {{ cuda11_cuda_profiler_api_run_version }}
Expand All @@ -235,9 +281,12 @@ outputs:
- {{ pin_compatible('cuda-version', max_pin='x', min_pin='x') }}
{% if cuda_major == "11" %}
- cudatoolkit
{% else %}
- cuda-cudart
- libcublas
{% endif %}
- libraft ={{ minor_version }}
- libcuvs ={{ minor_version }}
- {{ pin_subpackage('libcuvs', exact=True) }}
about:
home: https://rapids.ai/
license: Apache-2.0
Expand Down

0 comments on commit 808ff83

Please sign in to comment.