Fp64 on the CPU (#1843)

* add fp64 data type * clean build * update docs * fix bug
2025-12-16 01:49:05 +08:00 · 2025-02-07 15:52:22 -08:00
parent 1a1b2108ec
commit 1c0c118f7c
32 changed files with 438 additions and 65 deletions
--- a/mlx/backend/cpu/binary_two.h
+++ b/mlx/backend/cpu/binary_two.h
@@ -205,6 +205,9 @@ void binary(
    case float32:
      binary_op<float>(a, b, outputs, op);
      break;
+    case float64:
+      binary_op<double>(a, b, outputs, op);
+      break;
    case bfloat16:
      binary_op<bfloat16_t>(a, b, outputs, op);
      break;