diff --git a/mlx/backend/cpu/scan.cpp b/mlx/backend/cpu/scan.cpp index 199dbab35..33addd161 100644 --- a/mlx/backend/cpu/scan.cpp +++ b/mlx/backend/cpu/scan.cpp @@ -330,7 +330,8 @@ void Scan::eval_cpu(const std::vector& inputs, array& out) { reduce_type_, in, out, axis_, reverse_, inclusive_); break; case complex64: - throw std::runtime_error("Scan ops do not support complex types yet"); + scan_dispatch( + reduce_type_, in, out, axis_, reverse_, inclusive_); break; } });