zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-11-04 10:38:10 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Awni Hannun
2024-01-14 15:41:56 -08:00
parent 7a753335d7
commit 9d70412a91
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mlx/compile.cpp
View File

@@ -334,7 +334,7 @@ std::vector<array> compile_replace(
std::function<std::vector<array>(const std::vector<array>&)> compile( std::function<std::vector<array>(const std::vector<array>&)> compile(
const std::function<std::vector<array>(const std::vector<array>&)>& fun, const std::function<std::vector<array>(const std::vector<array>&)>& fun,
size_t fun_id) { size_t fun_id) {
return [&fun, fun_id](const std::vector<array>& inputs) { return [fun, fun_id](const std::vector<array>& inputs) {
// Find a cache entry with the correct inputs // Find a cache entry with the correct inputs
auto& entry = compiler_cache().find(fun_id, inputs); auto& entry = compiler_cache().find(fun_id, inputs);