Fix MPI distributed tests with CUDA backend (#2775)

2025-12-16 01:49:05 +08:00 · 2025-11-16 07:12:18 +09:00
parent 1bf605d56d
commit 9ac7dbe877
2 changed files with 10 additions and 5 deletions
--- a/mlx/backend/cuda/utils.cpp
+++ b/mlx/backend/cuda/utils.cpp
@@ -60,7 +60,7 @@ const char* dtype_to_cuda_type(const Dtype& dtype) {
    case float64:
      return "double";
    case complex64:
-      return "complex64_t";
+      return "mlx::core::cu::complex64_t";
    default:
      return "unknown";
  }