mirror of
https://github.com/ml-explore/mlx.git
synced 2025-12-16 01:49:05 +08:00
Compare commits
2 Commits
d4f4ff3c5e
...
7a6adda1e6
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
7a6adda1e6 | ||
|
|
1a9f820af6 |
@@ -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();
|
Stream s = arr.primitive().stream();
|
||||||
recurse(arr, 0, s, arr.shape());
|
recurse(arr, 0, s, arr.shape());
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -4,7 +4,7 @@
|
|||||||
|
|
||||||
#define MLX_VERSION_MAJOR 0
|
#define MLX_VERSION_MAJOR 0
|
||||||
#define MLX_VERSION_MINOR 29
|
#define MLX_VERSION_MINOR 29
|
||||||
#define MLX_VERSION_PATCH 1
|
#define MLX_VERSION_PATCH 2
|
||||||
#define MLX_VERSION_NUMERIC \
|
#define MLX_VERSION_NUMERIC \
|
||||||
(100000 * MLX_VERSION_MAJOR + 1000 * MLX_VERSION_MINOR + MLX_VERSION_PATCH)
|
(100000 * MLX_VERSION_MAJOR + 1000 * MLX_VERSION_MINOR + MLX_VERSION_PATCH)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user