From e444e2cbc04133d77004d3017831916e1d1130ad Mon Sep 17 00:00:00 2001
From: Yury Popov <git@phoenix.dj>
Date: Sun, 20 Apr 2025 02:06:28 +0300
Subject: [PATCH] cpu: allow complex logaddexp / atan2

---
 mlx/backend/cpu/binary.cpp | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/mlx/backend/cpu/binary.cpp b/mlx/backend/cpu/binary.cpp
index dbdab6a06..35aa2a3e0 100644
--- a/mlx/backend/cpu/binary.cpp
+++ b/mlx/backend/cpu/binary.cpp
@@ -172,9 +172,12 @@ void binary_float(
       case bfloat16:
         binary_op<bfloat16_t, Op>(a, b, out, bopt);
         break;
+      case complex64:
+        binary_op<complex64_t, Op>(a, b, out, bopt);
+        break;
       default:
         throw std::runtime_error(
-            "[binary_float] Only supports non-complex floating point types.");
+            "[binary_float] Only supports floating point types.");
     }
   });
 }