From 42cc9cfbc7f468d2d052fc16c09b5da01198969f Mon Sep 17 00:00:00 2001
From: Awni Hannun <awni@apple.com>
Date: Fri, 11 Jul 2025 10:59:35 -0700
Subject: [PATCH] fix copy dispatch (#2360)

---
 mlx/backend/metal/copy.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mlx/backend/metal/copy.cpp b/mlx/backend/metal/copy.cpp
index 8123b793e..915fc69fd 100644
--- a/mlx/backend/metal/copy.cpp
+++ b/mlx/backend/metal/copy.cpp
@@ -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";
     }
   }