diff --git a/mlx/backend/cuda/primitives.cu b/mlx/backend/cuda/primitives.cu index d143f9fa5b..137dd3b3c9 100644 --- a/mlx/backend/cuda/primitives.cu +++ b/mlx/backend/cuda/primitives.cu @@ -122,7 +122,6 @@ NO_GPU_MULTI(CustomKernel) } // namespace fast namespace distributed { -// NO_GPU_MULTI(AllReduce) NO_GPU_MULTI(AllGather) NO_GPU_MULTI(Send) NO_GPU_MULTI(Recv)