Make sliceUpdate general (#2282)

* Make sliceUpdate general * fix
2025-12-16 01:49:05 +08:00 · 2025-06-12 16:48:54 -07:00
parent c2dd81a8aa
commit f5f65ef48c
4 changed files with 38 additions and 36 deletions
--- a/mlx/distributed/mpi/mpi.cpp
+++ b/mlx/distributed/mpi/mpi.cpp
@@ -225,6 +225,8 @@ struct MPIWrapper {
        return mpi_bfloat16_;
      case float64:
        return mpi_double_;
+      default:
+        throw std::runtime_error("Invalid type");
    }
  }