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 Packages Projects Releases Wiki Activity

Address comment

Browse Source
This commit is contained in:
Angelos Katharopoulos
2025-09-30 10:26:43 -07:00
parent 3a7ad1b65b
commit 0ac8835955
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
mlx/compile.cpp
View File

@@ -995,7 +995,7 @@ ArrayFnWithExtra compile(
// At this point we must have a tape, now replace the placeholders // At this point we must have a tape, now replace the placeholders
// with real arrays that can be evaluated // with real arrays that can be evaluated
return std::pair<std::vector<array>, std::shared_ptr<void>>{ return ArraysAndExtra{
compile_replace( compile_replace(
entry.tape, entry.inputs, entry.outputs, inputs, shapeless), entry.tape, entry.inputs, entry.outputs, inputs, shapeless),
entry.extra}; entry.extra};
@@ -1017,8 +1017,7 @@ std::function<std::vector<array>(const std::vector<array>&)> compile(
ArrayFnWithExtra fun_with_extra = ArrayFnWithExtra fun_with_extra =
[fun = std::move(fun)](const std::vector<array>& inputs) { [fun = std::move(fun)](const std::vector<array>& inputs) {
return std::pair<std::vector<array>, std::shared_ptr<void>>{ return ArraysAndExtra{fun(inputs), nullptr};
fun(inputs), nullptr};
}; };
auto compiled_fun = compile( auto compiled_fun = compile(