From bc72e5bd3291a334f4ee84cae03fdac6ba6b8095 Mon Sep 17 00:00:00 2001 From: Yineng Zhang Date: Sun, 9 Feb 2025 22:57:11 +0800 Subject: [PATCH] add cuda graph capture failure possible solution (#3430) --- python/sglang/srt/model_executor/cuda_graph_runner.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/sglang/srt/model_executor/cuda_graph_runner.py b/python/sglang/srt/model_executor/cuda_graph_runner.py index 978a772f1..fc892ddf0 100644 --- a/python/sglang/srt/model_executor/cuda_graph_runner.py +++ b/python/sglang/srt/model_executor/cuda_graph_runner.py @@ -237,6 +237,7 @@ class CudaGraphRunner: "1. disable cuda graph by --disable-cuda-graph\n" "2. set --mem-fraction-static to a smaller value (e.g., 0.8 or 0.7)\n" "3. disable torch compile by not using --enable-torch-compile\n" + "4. set --cuda-graph-max-bs to a smaller value (e.g., 32)\n" "Open an issue on GitHub https://github.com/sgl-project/sglang/issues/new/choose \n" )