Add libcublas-dev, libcusolver-dev, libcusparse-dev to host deps

Needed to compile flash-attn on CUDA 12.0 in conda-forge.
weiji14 · May 4, 2024 · 4d8b37c · 4d8b37c
1 parent aa17a2c
commit 4d8b37c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/recipes/flash-attn/meta.yaml b/recipes/flash-attn/meta.yaml
@@ -22,6 +22,9 @@ requirements:
     - {{ compiler('cuda') }}
     - {{ stdlib('c') }}
   host:
+    - libcublas-dev                   # [(cuda_compiler_version or "").startswith("12")]
+    - libcusolver-dev                 # [(cuda_compiler_version or "").startswith("12")]
+    - libcusparse-dev                 # [(cuda_compiler_version or "").startswith("12")]
     - packaging
     - pip
     - psutil