Force cudaGraphExec reinstantiation when clusters are used (#2813)

Co-authored-by: Awni Hannun <awni@apple.com>
2025-12-16 01:49:05 +08:00 · 2025-11-22 15:43:49 -05:00
parent 5b0f047226
commit 3e05cea9f8
3 changed files with 91 additions and 52 deletions
--- a/mlx/backend/cuda/utils.cpp
+++ b/mlx/backend/cuda/utils.cpp
@@ -5,6 +5,7 @@
 #include "mlx/dtype_utils.h"

 #include <fmt/format.h>
+#include <vector>

 namespace mlx::core {