From 676fad064f3bccd1dfcd61658acc9badd4abd9ec Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Wed, 3 Jun 2026 21:45:15 +0000
Subject: [PATCH] Fix: Add out= parameter to run_fused_swiglu_grouped_gemm
 signature

---
 dsv4/ops/gemm_runner.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/dsv4/ops/gemm_runner.py b/dsv4/ops/gemm_runner.py
index 2b4f615c..5d0501e7 100644
--- a/dsv4/ops/gemm_runner.py
+++ b/dsv4/ops/gemm_runner.py
@@ -416,6 +416,7 @@ def run_fused_swiglu_grouped_gemm(
     swiglu_limit=0.0,
     mma_tiler_mn=(128, 128),
     cluster_shape_mn=(1, 1),
+    out=None,       # pre-allocated output buffer for CUDA graph capture
 ):
     """Run the fused SwiGLU NVFP4 scaled grouped GEMM.