Cuda bug fixes 2 (#2298)

* more bug fixes * more bug fixes * format
2025-12-16 01:49:05 +08:00 · 2025-06-16 13:14:46 -07:00
parent c552ff2451
commit bc53f8293f
11 changed files with 143 additions and 107 deletions
--- a/mlx/backend/cuda/utils.cpp
+++ b/mlx/backend/cuda/utils.cpp
@@ -31,6 +31,9 @@ const char* dtype_to_cuda_type(const Dtype& dtype) {
  if (dtype == bfloat16) {
    return "__nv_bfloat16";
  }
+  if (dtype == complex64) {
+    return "cuComplex";
+  }
 #define SPECIALIZE_DtypeToString(CPP_TYPE, DTYPE) \
  if (dtype == DTYPE) {                           \
    return #CPP_TYPE;                             \