From dd9af0114e62c11c4e1f78454ab37a9da4d158f8 Mon Sep 17 00:00:00 2001 From: Georgi Gerganov Date: Mon, 16 Feb 2026 18:13:21 +0200 Subject: [PATCH] cont : add stream capture check Co-authored-by: Oliver Simons --- ggml/src/ggml-cuda/ggml-cuda.cu | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/ggml/src/ggml-cuda/ggml-cuda.cu b/ggml/src/ggml-cuda/ggml-cuda.cu index 1861f19688..de5a0bbd0f 100644 --- a/ggml/src/ggml-cuda/ggml-cuda.cu +++ b/ggml/src/ggml-cuda/ggml-cuda.cu @@ -2307,7 +2307,9 @@ static void ggml_cuda_mul_mat_id(ggml_backend_cuda_context & ctx, ggml_tensor * } // note: this path should not be reached when recording CUDA graphs, because it requires stream synchronization - + cudaStreamCaptureStatus capture_status; + CUDA_CHECK(cudaStreamIsCapturing(stream, &capture_status)); + GGML_ASSERT(capture_status == cudaStreamCaptureStatusNone); cudaStream_t stream = ctx.stream(); GGML_ASSERT(nb12 % nb11 == 0);