From a0558cd3afa4c9b1a78f4fbc1e09616e737f0255 Mon Sep 17 00:00:00 2001
From: Jagrit Digani <digani@apple.com>
Date: Fri, 14 Nov 2025 16:03:34 -0800
Subject: [PATCH] Fix routing for complex

---
 mlx/backend/metal/matmul.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mlx/backend/metal/matmul.cpp b/mlx/backend/metal/matmul.cpp
index 540fbce02..9f435f73c 100644
--- a/mlx/backend/metal/matmul.cpp
+++ b/mlx/backend/metal/matmul.cpp
@@ -359,7 +359,7 @@ void steel_matmul_regular_axpby(
     float beta /* = 0.0f */) {
 #ifdef MLX_ENABLE_NAX
 
-  if (metal::is_nax_available() &&
+  if (metal::is_nax_available() && !issubdtype(a.dtype(), complexfloating) &&
       (a.dtype() != float32 || env::enable_tf32())) {
     return steel_matmul_regular_axpby_nax<CHECK_AB>(
         /* const Stream& s = */ s,