fix function pointer (#1865)

andresy · web-flow · commit 9733e1649645 · 2025-02-13T18:46:11.000-08:00
diff --git a/mlx/compile.cpp b/mlx/compile.cpp
@@ -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;
diff --git a/mlx/compile.h b/mlx/compile.h
@@ -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.

Original file line number	Diff line number	Diff line change
`@@ -922,7 +922,7 @@ std::function<std::vector<array>(const std::vector<array>&)> compile(`
`922`	`922`	`}`
`923`	`923`
`924`	`924`	`std::function<std::vector<array>(const std::vector<array>&)> compile(`
`925`		`- std::vector<array>(fun)(const std::vector<array>&),`
	`925`	`+ std::vector<array> (*fun)(const std::vector<array>&),`
`926`	`926`	`bool shapeless /* = false */) {`
`927`	`927`	`if (detail::skip_compile()) {`
`928`	`928`	`return fun;`