More shape type (#1705)

* more shape type * fix
2025-12-16 01:49:05 +08:00 · 2024-12-19 08:08:20 -08:00
parent f17536af9c
commit e03f0372b1
38 changed files with 260 additions and 258 deletions
--- a/mlx/io/load.cpp
+++ b/mlx/io/load.cpp
@@ -271,7 +271,7 @@ array load(std::shared_ptr<io::Reader> in_stream, StreamOrDevice s) {
  bool col_contiguous = header[34] == 'T';

  // Read array shape from header
-  std::vector<int> shape;
+  Shape shape;

  size_t st = header.find_last_of('(') + 1;
  size_t ed = header.find_last_of(')');