From 5a1a5d5ed16f69af7c3ce56dd94e4502661e1565 Mon Sep 17 00:00:00 2001 From: Awni Hannun Date: Mon, 5 May 2025 17:30:50 -0700 Subject: [PATCH] fix input coherent kernel launch (#2153) --- mlx/backend/metal/fence.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mlx/backend/metal/fence.cpp b/mlx/backend/metal/fence.cpp index d4a88d983..5abdf7309 100644 --- a/mlx/backend/metal/fence.cpp +++ b/mlx/backend/metal/fence.cpp @@ -138,7 +138,7 @@ void Fence::update(Stream stream, const array& x) { compute_encoder.set_compute_pipeline_state(kernel); compute_encoder.set_input_array(x, 0); compute_encoder.set_bytes(nthreads, 1); - compute_encoder.dispatch_threadgroups(group_dims, grid_dims); + compute_encoder.dispatch_threadgroups(grid_dims, group_dims); // Barrier on previous kernels compute_encoder.barrier();