zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-07-14 20:41:13 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix copy dispatch (#2360)

Browse Source
This commit is contained in:
Awni Hannun 2025-07-11 10:59:35 -07:00 committed by GitHub
parent 8347575ba1
commit 42cc9cfbc7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -86,7 +86,7 @@ void copy_gpu_inplace(
}
} else {
work_per_thread = get_work_per_thread(out.dtype(), out.data_size());
if (work_per_thread > 1) {
if (!large && work_per_thread > 1) {
kernel_name += "n";
}
}