Remove the kernel arg from get_launch_args (#2437)

2025-12-16 01:49:05 +08:00 · 2025-07-30 11:43:02 +09:00
parent 3adba92ebe
commit 254476718b
13 changed files with 83 additions and 125 deletions
--- a/mlx/backend/cuda/copy/copy_contiguous.cu
+++ b/mlx/backend/cuda/copy/copy_contiguous.cu
@@ -71,12 +71,7 @@ void copy_contiguous(
          kernel = cu::copy_v<InType, OutType, IdxT, N_READS>;
        }
        auto [num_blocks, block_dims] = get_launch_args(
-            kernel,
-            out.data_size(),
-            out.shape(),
-            out.strides(),
-            large(),
-            N_READS);
+            out.data_size(), out.shape(), out.strides(), large(), N_READS);
        encoder.add_kernel_node(
            kernel,
            num_blocks,