From 8917022deb609511778dc93c193480967e43b777 Mon Sep 17 00:00:00 2001 From: Awni Hannun Date: Wed, 2 Jul 2025 19:37:58 -0700 Subject: [PATCH] fix graphs for older cuda (#2328) --- mlx/backend/cuda/device.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mlx/backend/cuda/device.cpp b/mlx/backend/cuda/device.cpp index fff752fe5..4129563af 100644 --- a/mlx/backend/cuda/device.cpp +++ b/mlx/backend/cuda/device.cpp @@ -54,8 +54,8 @@ void Device::make_current() { CommandEncoder::CaptureContext::CaptureContext(CommandEncoder& enc) : enc(enc) { CHECK_CUDA_ERROR(cudaGraphCreate(&graph, 0)); - CHECK_CUDA_ERROR(cudaStreamBeginCaptureToGraph( - enc.stream(), graph, NULL, NULL, 0, cudaStreamCaptureModeGlobal)); + CHECK_CUDA_ERROR( + cudaStreamBeginCapture(enc.stream(), cudaStreamCaptureModeGlobal)); } CommandEncoder::CaptureContext::~CaptureContext() {