zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix (#1566)

Browse Source
This commit is contained in:
Awni Hannun
2024-11-06 17:10:33 -08:00
committed by GitHub
parent 726dbd9267
commit 9a3842a2d9
4 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mlx/backend/metal/scaled_dot_product_attention.cpp
View File

@@ -139,7 +139,7 @@ void sdpa_full_self_attention_metal(
MTL::Size grid_dims = MTL::Size(1, tm, batch_size_out);
MTL::Size group_dims = MTL::Size(32, wm, wn);
compute_encoder->dispatchThreadgroups(grid_dims, group_dims);
compute_encoder.dispatchThreadgroups(grid_dims, group_dims);
}
void sdpa_vector(