zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-19 00:04:41 +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
1 changed files with 1 additions and 1 deletions
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";
} }
} }