Return empty array when repeats is 0 in mx.repeat (#681)

* Return empty array when repeats is 0 * Add test case for repeats = 0
2025-09-19 02:38:09 +08:00 · 2024-02-14 02:49:31 +01:00
parent 2fdc2462c3
commit 0c65517e91
2 changed files with 3 additions and 1 deletions
--- a/mlx/ops.cpp
+++ b/mlx/ops.cpp
@@ -774,7 +774,7 @@ array repeat(const array& arr, int repeats, int axis, StreamOrDevice s) {
  }

  if (repeats == 0) {
-    return array({}, arr.dtype());
+    return array(std::initializer_list<int>{}, arr.dtype());
  }

  if (repeats == 1) {