Cleanup using namespace alias

benchmarks/cpp/single_ops.cpp

@@ -193,8 +193,8 @@ void time_reductions() {
   auto argmin_along_1 = [&a]() { return mx::argmin(a, 1, false); };
   TIME(argmin_along_1);
 
-  auto indices = mlx::core::array({1});
-  auto updates = mlx::core::reshape(mlx::core::array({NAN}), {1, 1, 1});
+  auto indices = mx::array({1});
+  auto updates = mx::reshape(mx::array({NAN}), {1, 1, 1});
   std::vector<int> axes{0};
   auto b = scatter(a, {indices}, updates, axes);
   mx::eval(b);

mlx/backend/metal/kernels/reduction/ops.h

@@ -186,7 +186,12 @@ struct Max {
   DEFINE_SIMD_REDUCE()
 
   template <typename T>
-  T simd_reduce_impl(T val) {
+  metal::enable_if_t<metal::is_integral_v<T>, T> simd_reduce_impl(T val) {
+    return simd_max(val);
+  }
+
+  template <typename T>
+  metal::enable_if_t<!metal::is_integral_v<T>, T> simd_reduce_impl(T val) {
     if (simd_any(val != val)) {
       return static_cast<T>(NAN);
     }