From 70ffaa50d29347aaff47531867f31b6bd1d38278 Mon Sep 17 00:00:00 2001
From: Ronan Collobert <collobert@apple.com>
Date: Wed, 13 Nov 2024 13:51:37 -0800
Subject: [PATCH] be more relaxed on OpenMPI version

---
 CMakeLists.txt | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index d8c9d5ba9..6c6cb8996 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -168,11 +168,12 @@ endif()
 find_package(MPI)
 if(MPI_FOUND)
   execute_process(
-    COMMAND zsh "-c" "mpirun --version"
+    COMMAND zsh "-c" "${MPIEXEC_EXECUTABLE} --version"
     OUTPUT_VARIABLE MPI_VERSION
     ERROR_QUIET)
-  if(${MPI_VERSION} MATCHES ".*Open MPI.*")
+  if(${MPI_VERSION} MATCHES ".*Open MPI.*" OR ${MPI_VERSION} MATCHES ".*OpenRTE.*")
     target_include_directories(mlx PRIVATE ${MPI_INCLUDE_PATH})
+    target_link_libraries(mlx PRIVATE ${MPI_CXX_LIBRARIES})
   elseif(MPI_VERSION STREQUAL "")
     set(MPI_FOUND FALSE)
     message(