Skip to content

Commit

Permalink
Use pin_compatible to ensure that lower CTKs can be used (#1462)
Browse files Browse the repository at this point in the history
The runtime requirement for the CTK is weaker than the build-time requirement. We allow installing raft into environments with an older CTK (any 11.x) than the build requirement (11.8)

Authors:
  - Vyas Ramasubramani (https://github.com/vyasr)

Approvers:
  - Corey J. Nolet (https://github.com/cjnolet)
  - Divye Gala (https://github.com/divyegala)
  - AJ Schmidt (https://github.com/ajschmidt8)
  - Bradley Dice (https://github.com/bdice)

URL: #1462
  • Loading branch information
vyasr authored Apr 25, 2023
1 parent b00d2f7 commit 1866120
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 2 deletions.
7 changes: 6 additions & 1 deletion conda/recipes/libraft/meta.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -52,6 +52,9 @@ outputs:
host:
- librmm ={{ minor_version }}
- cudatoolkit {{ cuda_version }}
run:
- {{ pin_compatible('cudatoolkit', max_pin='x', min_pin='x') }}
- librmm ={{ minor_version }}
about:
home: https://rapids.ai/
license: Apache-2.0
Expand All @@ -68,7 +71,6 @@ outputs:
run:
- {{ pin_subpackage('libraft-headers-only', exact=True) }}
- cuda-profiler-api {{ cuda_profiler_api_run_version }}
- cudatoolkit {{ cuda_version }}
- librmm ={{ minor_version }}
- libcublas {{ libcublas_run_version }}
- libcublas-dev {{ libcublas_run_version }}
Expand Down Expand Up @@ -101,6 +103,7 @@ outputs:
- sysroot_{{ target_platform }} {{ sysroot_version }}
host:
- {{ pin_subpackage('libraft-headers', exact=True) }}
- cudatoolkit {{ cuda_version }}
- cuda-profiler-api {{ cuda_profiler_api_host_version }}
- libcublas {{ libcublas_host_version }}
- libcublas-dev {{ libcublas_host_version }}
Expand Down Expand Up @@ -135,6 +138,7 @@ outputs:
- sysroot_{{ target_platform }} {{ sysroot_version }}
host:
- {{ pin_subpackage('libraft', exact=True) }}
- cudatoolkit {{ cuda_version }}
- cuda-profiler-api {{ cuda_profiler_api_host_version }}
- gmock {{ gtest_version }}
- gtest {{ gtest_version }}
Expand Down Expand Up @@ -200,6 +204,7 @@ outputs:
- sysroot_{{ target_platform }} {{ sysroot_version }}
host:
- {{ pin_subpackage('libraft', exact=True) }}
- cudatoolkit {{ cuda_version }}
- libcublas {{ libcublas_host_version }}
- libcublas-dev {{ libcublas_host_version }}
- glog {{ glog_version }}
Expand Down
2 changes: 1 addition & 1 deletion cpp/include/raft/distance/detail/kernels/gram_matrix.cuh
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@
#include <raft/core/device_resources.hpp>
#include <raft/distance/distance.cuh>
#include <raft/distance/distance_types.hpp>
//#include <raft/sparse/detail/cusparse_wrappers.h>
// #include <raft/sparse/detail/cusparse_wrappers.h>
#include <raft/sparse/distance/distance.cuh>
#include <raft/sparse/linalg/spmm.cuh>

Expand Down

0 comments on commit 1866120

Please sign in to comment.