diff --git a/recipes/flash-attn/meta.yaml b/recipes/flash-attn/meta.yaml index 745d0469bb9ed..dc14b478a67a9 100644 --- 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