zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-07-15 04:51: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 { } else {
work_per_thread = get_work_per_thread(out.dtype(), out.data_size()); 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"; kernel_name += "n";
} }
} }