Skip to content

Commit

Permalink
Remove faiss dep for CUDA 12
Browse files Browse the repository at this point in the history
  • Loading branch information
vyasr committed Apr 6, 2023
1 parent 45f9bf0 commit 7e66ff2
Showing 1 changed file with 6 additions and 4 deletions.
10 changes: 6 additions & 4 deletions conda/recipes/libcuml/meta.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -62,6 +62,9 @@ requirements:
- libcusolver-dev {{ cuda11_libcusolver_host_version }}
- libcusparse {{ cuda11_libcusparse_host_version }}
- libcusparse-dev {{ cuda11_libcusparse_host_version }}
# Ignore faiss in CUDA 12 for now
- libfaiss>=1.7.1
- faiss-proc=*=cuda
{% else %}
- cuda-cudart {{ cuda_version }}
- cuda-cudart-dev {{ cuda_version }}
Expand All @@ -81,8 +84,6 @@ requirements:
- libraft ={{ minor_version }}
- libraft-headers ={{ minor_version }}
- treelite {{ treelite_version }}
- libfaiss>=1.7.1
- faiss-proc=*=cuda

outputs:
- name: libcuml
Expand All @@ -106,6 +107,9 @@ outputs:
- libcurand {{ cuda11_libcurand_run_version }}
- libcusparse {{ cuda11_libcusparse_run_version }}
- libcusparse-dev {{ cuda11_libcusparse_run_version }}
# Ignore faiss in CUDA 12 for now
- libfaiss>=1.7.1
- faiss-proc=*=cuda
{% else %}
- cuda-cudart {{ cuda_spec }}
- cuda-cudart-dev {{ cuda_spec }}
Expand All @@ -121,8 +125,6 @@ outputs:
- libraft ={{ minor_version }}
- libraft-headers ={{ minor_version }}
- treelite {{ treelite_version }}
- libfaiss>=1.7.1
- faiss-proc=*=cuda
about:
home: https://rapids.ai/
license: Apache-2.0
Expand Down

0 comments on commit 7e66ff2

Please sign in to comment.