Fix stale CUDA graph benchmark and docs refs (#28041)

2026-06-13 21:51:42 -07:00
parent 5da3b37a9d
commit 91c63aeb4d
4 changed files with 11 additions and 11 deletions
@@ -7,10 +7,10 @@ from torch.nn import functional as F
 from transformers import AutoConfig

 from sglang.benchmark.bench_utils import run_bench
+from sglang.srt.compilation.torch_compile_decoration import set_torch_compile_config
 from sglang.srt.layers.moe.moe_runner.triton_utils.fused_moe import (
    fused_moe as fused_moe_triton,
 )
-from sglang.srt.model_executor.cuda_graph_runner import set_torch_compile_config


 def get_model_config(model_name: str, tp_size: int):