[CI] Initialize ubatch runner in cudagraph unit test (#56808)
Signed-off-by: Kevin Luu <51931015+khluu@users.noreply.github.com> Co-authored-by: OpenAI Codex <codex@openai.com>
Kevin H. Luu committed
Sep 14, 2026 at 11:37 UTC
56ca9904ce76975fe3aad28d1af7af961a07f054
1 file changed
+1
tests/v1/cudagraph/test_breakable_cudagraph.py
+1
@@ -43,6 +43,7 @@ def test_piecewise_capture_builds_fresh_metadata_for_both_passes():
43
manager._capture_descs = {CUDAGraphMode.PIECEWISE: [desc]}
44
manager._graphs_captured = False
45
manager.use_breakable_cg = True
46
+ manager.ubatch_runner = None
47
48
create_calls = []
49
forward_calls = []