zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-25 01:41:17 +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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions
Show Stats 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::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 */) { bool shapeless /* = false */) {
if (detail::skip_compile()) { if (detail::skip_compile()) {
return fun; 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); bool shapeless = false);
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); bool shapeless = false);
// Convert capture-less lambdas to function pointers. // Convert capture-less lambdas to function pointers.