From 9733e16496455d6fee204d70d15621df3fe20b95 Mon Sep 17 00:00:00 2001
From: Ronan Collobert <collobert@apple.com>
Date: Thu, 13 Feb 2025 18:46:11 -0800
Subject: [PATCH] fix function pointer (#1865)

---
 mlx/compile.cpp | 2 +-
 mlx/compile.h   | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/mlx/compile.cpp b/mlx/compile.cpp
index 8b7f24c91..ced958b13 100644
--- 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
index 90b6a9c2a..a076cfbca 100644
--- 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.