diff --git a/mlx/backend/cpu/binary.cpp b/mlx/backend/cpu/binary.cpp index dbdab6a06a..35aa2a3e0d 100644 --- a/mlx/backend/cpu/binary.cpp +++ b/mlx/backend/cpu/binary.cpp @@ -172,9 +172,12 @@ void binary_float( case bfloat16: binary_op(a, b, out, bopt); break; + case complex64: + binary_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."); } }); }