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

allow compiling lambdas in C++ (#1650)

Browse Source 
* allow compiling lambdas in C++

* fix test

* more tests

* auto detect capture-less lambda
This commit is contained in:
Awni Hannun
2024-12-06 13:13:21 -08:00
committed by GitHub
parent fd3377dd1f
commit 69a2991614
4 changed files with 115 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mlx/compile_impl.h
View File

@@ -9,7 +9,7 @@ namespace mlx::core::detail {
// This is not part of the general C++ API as calling with a bad id is a bad
// idea.
std::function<std::vector<array>(const std::vector<array>&)> compile(
const std::function<std::vector<array>(const std::vector<array>&)>& fun,
std::function<std::vector<array>(const std::vector<array>&)> fun,
std::uintptr_t fun_id,
bool shapeless = false,
std::vector<uint64_t> constants = {});