zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-01 04:24:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix cudnn routing

Browse Source
This commit is contained in:
Jagrit Digani
2025-08-06 15:05:58 -07:00
parent c66b76a8c8
commit 99d8de8445
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
mlx/backend/cuda/scaled_dot_product_attention.cu
View File

@@ -1068,8 +1068,7 @@ void ScaledDotProductAttention::eval_gpu(
flags);
}
// return sdpa_vector_fallback(s, encoder, q, k, v, scale_, o, do_causal_);
return sdpa_cudnn(s, encoder, q, k, v, scale_, o, do_causal_);
return sdpa_vector_fallback(s, encoder, q, k, v, scale_, o, do_causal_);
}
// Full attention mode