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

fix function pointer (#1865)

Browse Source
This commit is contained in:
Ronan Collobert
2025-02-13 18:46:11 -08:00
committed by GitHub
parent 7f2d1024f3
commit 9733e16496
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mlx/compile.cpp
View File

@@ -922,7 +922,7 @@ std::function<std::vector<array>(const std::vector<array>&)> compile(
}
std::function<std::vector<array>(const std::vector<array>&)> compile(
std::vector<array>(fun)(const std::vector<array>&),
std::vector<array> (*fun)(const std::vector<array>&),
bool shapeless /* = false */) {
if (detail::skip_compile()) {
return fun;

2
mlx/compile.h
View File

@@ -14,7 +14,7 @@ std::function<std::vector<array>(const std::vector<array>&)> compile(
bool shapeless = false);
std::function<std::vector<array>(const std::vector<array>&)> compile(
std::vector<array>(fun)(const std::vector<array>&),
std::vector<array> (*fun)(const std::vector<array>&),
bool shapeless = false);
// Convert capture-less lambdas to function pointers.