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

Ensure the same behaviour as before

Browse Source
This commit is contained in:
Angelos Katharopoulos
2025-09-29 23:39:04 -07:00
parent 7cdb155d17
commit 6faf593775
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
mlx/compile.cpp
View File

@@ -1007,6 +1007,14 @@ std::function<std::vector<array>(const std::vector<array>&)> compile(
std::uintptr_t fun_id,
bool shapeless /* = false */,
std::vector<uint64_t> constants /* = {} */) {
if (skip_compile()) {
return fun;
}
if (!fun) {
throw std::invalid_argument(
"[compile] Cannot compile a function without a target.");
}
ArrayFnWithExtra fun_with_extra =
[fun = std::move(fun)](const std::vector<array>& inputs) {
return std::pair<std::vector<array>, std::shared_ptr<void>>{