Use SmallVector for shapes and strides (#2454)

* Use SmallVector for shapes and strides * Convert SmallVector to tuple
2025-08-05 09:41:03 +09:00
parent 7d86a5c108
commit 828c5f1137
30 changed files with 738 additions and 102 deletions
@@ -9,7 +9,7 @@

 #include "mlx/distributed/distributed.h"
 #include "mlx/distributed/ops.h"
-
+#include "python/src/small_vector.h"
 #include "python/src/utils.h"

 namespace mx = mlx::core;