[Memory] Borrow CUDA graph pool storage for EAGLE sampling (#35375)

Co-authored-by: cctry <cctry@fb.com>
2026-08-18 16:19:41 -07:00
parent 7ebaa98f81
commit 37c09ff3d8
11 changed files with 783 additions and 184 deletions
@@ -33,7 +33,10 @@ _KITS = (


 class _Core(Eagle3Base):
-    env_overrides = ((envs.SGLANG_ENABLE_STRICT_MEM_CHECK_DURING_BUSY, 1),)
+    env_overrides = (
+        (envs.SGLANG_ENABLE_STRICT_MEM_CHECK_DURING_BUSY, 1),
+        (envs.SGLANG_ENABLE_GRAPH_POOL_BORROW, 1),
+    )


 class TestEagle3Overlap(_Core, *_KITS):