From fa56bf2feb65ce1f47420647a861f8f73074f254 Mon Sep 17 00:00:00 2001 From: Angelos Katharopoulos Date: Tue, 19 Aug 2025 14:18:21 -0700 Subject: [PATCH] Remove completion handler from custom kernel --- mlx/backend/cuda/custom_kernel.cpp | 1 - 1 file changed, 1 deletion(-) diff --git a/mlx/backend/cuda/custom_kernel.cpp b/mlx/backend/cuda/custom_kernel.cpp index 5ce9d3097..704567ffa 100644 --- a/mlx/backend/cuda/custom_kernel.cpp +++ b/mlx/backend/cuda/custom_kernel.cpp @@ -374,7 +374,6 @@ void CustomKernel::eval_gpu( } }); encoder.add_kernel_node(kernel, grid, block, shared_memory_, args.args()); - encoder.add_completed_handler([args = std::move(args)]() {}); } } // namespace mlx::core::fast