zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

WIP (metal)

Browse Source
This commit is contained in:
Ronan Collobert
2025-10-31 16:24:21 -07:00
parent ac75c87fd7
commit 8d13a0bc6b
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
mlx/backend/metal/primitives.cpp
View File

@@ -26,6 +26,7 @@ void arange_set_scalars(T start, T next, metal::CommandEncoder& enc) {
void Arange::eval_gpu(const std::vector<array>& inputs, array& out) { void Arange::eval_gpu(const std::vector<array>& inputs, array& out) {
assert(inputs.size() == 0); assert(inputs.size() == 0);
(void)inputs;
out.set_data(allocator::malloc(out.nbytes())); out.set_data(allocator::malloc(out.nbytes()));
if (out.size() == 0) { if (out.size() == 0) {
return; return;