Skip to content

GitLab

Unverified Commit 1646149a authored by Yineng Zhang's avatar Yineng Zhang Committed by GitHub
Browse files

fix draft cuda graph capture failure (#3431)

parent bc72e5bd
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
python/sglang/srt/speculative/eagle_draft_cuda_graph_runner.py
View file @ 1646149a
...@@ -85,6 +85,7 @@ class EAGLEDraftCudaGraphRunner: ...@@ -85,6 +85,7 @@ class EAGLEDraftCudaGraphRunner:
"1. disable cuda graph by --disable-cuda-graph\n" "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" "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" "3. disable torch compile by not using --enable-torch-compile\n"
"4. specify --dtype to the same dtype (e.g. bfloat16)\n"
"Open an issue on GitHub https://github.com/sgl-project/sglang/issues/new/choose \n" "Open an issue on GitHub https://github.com/sgl-project/sglang/issues/new/choose \n"
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment