From 70f2baf39f17efbfc0b698553bf59bc9b8d7146d Mon Sep 17 00:00:00 2001
From: Anastasiia Filippova <anastasiia.filippovaa@gmail.com>
Date: Mon, 16 Jun 2025 19:11:28 +0200
Subject: [PATCH] Removed commented nogpu for all_reduce

---
 mlx/backend/cuda/primitives.cu | 1 -
 1 file changed, 1 deletion(-)

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)