Compile stride bug (#812)

* fix compile stride bug * revert sdpa fix * fix cpu * fix bug with simplifying outputs
2025-12-16 01:49:05 +08:00 · 2024-03-11 06:31:31 -07:00
parent a4d290adb9
commit 7c441600fe
9 changed files with 58 additions and 12 deletions
--- a/mlx/array.h
+++ b/mlx/array.h
@@ -339,6 +339,13 @@ class array {

  void copy_shared_buffer(const array& other);

+  void move_shared_buffer(
+      array other,
+      const std::vector<size_t>& strides,
+      Flags flags,
+      size_t data_size,
+      size_t offset = 0);
+
  void move_shared_buffer(array other);

  void overwrite_descriptor(const array& other) {