Fix typo in simd_reduce_mean

microsoft · Feb 27, 2024 · b601f4e · b601f4e
1 parent d7d12d9
commit b601f4e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mx/simd_ops.py b/mx/simd_ops.py
@@ -524,7 +524,7 @@ def simd_reduce_mean(in1, dim=None, keepdim=False, mx_specs=None):
     mx_assert_test(mx_specs)
     if mx_specs is None:
         if dim is None:
-            return torch_sum(in1)
+            return torch_mean(in1)
         else:
             return torch_mean(in1, dim, keepdim=keepdim)