diff --git a/mlx/backend/cuda/conv.cpp b/mlx/backend/cuda/conv.cpp index 4a2917254..fd3bc1118 100644 --- a/mlx/backend/cuda/conv.cpp +++ b/mlx/backend/cuda/conv.cpp @@ -147,6 +147,7 @@ bool execute_plan( return false; } + encoder.add_completed_handler([plan = std::move(plan)]() {}); encoder.add_temporary(workspace); return true; }