Update aten/src/ATen/native/mps/operations/ReduceOps.mm

malfet · web-flow · commit 23bb423ed8bb · 2025-03-23T20:05:49.000-07:00
diff --git a/aten/src/ATen/native/mps/operations/ReduceOps.mm b/aten/src/ATen/native/mps/operations/ReduceOps.mm
@@ -1559,10 +1559,6 @@ static void median_out_mps_common(const Tensor& input_t,
     return;
   }
 
-  if (values.numel() == 0) {
-    return;
-  }
-
   if (input_t.numel() == 1 && input_t.dim() == 0) {
     values.fill_(input_t);
     indices.fill_(0);
