Gather mm new kernel and small refactoring (#2040)

2025-12-16 01:49:05 +08:00 · 2025-04-14 16:37:36 -07:00
parent e9e268336b
commit 99eefd2ec0
23 changed files with 1260 additions and 378 deletions
--- a/mlx/backend/common/broadcasting.h
+++ b/mlx/backend/common/broadcasting.h
@@ -0,0 +1,11 @@
+// Copyright © 2024 Apple Inc.
+
+#pragma once
+
+#include "mlx/array.h"
+
+namespace mlx::core {
+
+void broadcast(const array& in, array& out);
+
+} // namespace mlx::core