From 25814a94589070f1d99c439413928c7a872538cf Mon Sep 17 00:00:00 2001
From: Angelos Katharopoulos <a_katharopoulos@apple.com>
Date: Fri, 21 Mar 2025 13:36:26 -0700
Subject: [PATCH] Disable mpi on version mismatch (#1989)

---
 mlx/distributed/mpi/mpi.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/mlx/distributed/mpi/mpi.cpp b/mlx/distributed/mpi/mpi.cpp
index 77b346037..ed0d66b3f 100644
--- a/mlx/distributed/mpi/mpi.cpp
+++ b/mlx/distributed/mpi/mpi.cpp
@@ -69,6 +69,7 @@ struct MPIWrapper {
     if (version.find("Open MPI") == std::string::npos) {
       std::cerr << "[mpi] MPI found but it does not appear to be Open MPI."
                 << "MLX requires Open MPI but this is " << version << std::endl;
+      libmpi_handle_ = nullptr;
       return;
     }