From 9239808225678e1572f587a6a4abfcf0d0fa6f9f Mon Sep 17 00:00:00 2001
From: Harrison Powers <harrisonpowers@gmail.com>
Date: Mon, 6 Apr 2026 15:32:16 -0400
Subject: [PATCH] Fix CMake finding wrong Python during pip install (#3375)

---
 setup.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/setup.py b/setup.py
index 04a905e1..d5172466 100644
--- a/setup.py
+++ b/setup.py
@@ -5,6 +5,7 @@ import os
 import platform
 import re
 import subprocess
+import sys
 from functools import partial
 from pathlib import Path
 
@@ -91,6 +92,7 @@ class CMakeBuild(build_ext):
             f"-DCMAKE_INSTALL_PREFIX={install_prefix}",
             f"-DMLX_PYTHON_BINDINGS_OUTPUT_DIRECTORY={pybind_out_dir}",
             f"-DCMAKE_BUILD_TYPE={cfg}",
+            f"-DPython_EXECUTABLE={sys.executable}",
             "-DMLX_BUILD_PYTHON_BINDINGS=ON",
             "-DMLX_BUILD_TESTS=OFF",
             "-DMLX_BUILD_BENCHMARKS=OFF",