Compare commits

...

2 Commits

Author SHA1 Message Date
Angelos Katharopoulos
7a6adda1e6 Bump the version (#2627) 2025-09-26 15:15:28 -07:00
Angelos Katharopoulos
1a9f820af6 Compiled should not end in broadcast (#2622) 2025-09-26 13:36:09 -07:00
2 changed files with 6 additions and 2 deletions

View File

@@ -727,7 +727,11 @@ void compile_fuse(
}
};
if (arr.has_primitive()) {
// This will be the result of the fused operation so it needs
// a) to not be already computed ie have a primitive
// b) that primitive to not be a broadcast since it will unnecessarily
// cast to a contiguous array potentially blowing up memory
if (arr.has_primitive() && !is_broadcast(arr.primitive())) {
Stream s = arr.primitive().stream();
recurse(arr, 0, s, arr.shape());
}

View File

@@ -4,7 +4,7 @@
#define MLX_VERSION_MAJOR 0
#define MLX_VERSION_MINOR 29
#define MLX_VERSION_PATCH 1
#define MLX_VERSION_PATCH 2
#define MLX_VERSION_NUMERIC \
(100000 * MLX_VERSION_MAJOR + 1000 * MLX_VERSION_MINOR + MLX_VERSION_PATCH)