support using fa4 on deepseek on blackwell (#9928)

This commit is contained in:
cicirori
2025-09-17 07:16:06 +08:00
committed by GitHub
parent 311de47bb7
commit a2f7218a2e
7 changed files with 136 additions and 0 deletions

View File

@@ -666,6 +666,13 @@ def _set_envs_and_config(server_args: ServerArgs):
if os.environ.get("TRTLLM_ENABLE_PDL", "1") != "0":
os.environ["TRTLLM_ENABLE_PDL"] = "1"
if os.environ.get("CUTE_DSL_LOG_LEVEL") is None:
# Default to warning level, to avoid too many logs
os.environ["CUTE_DSL_LOG_LEVEL"] = "30"
if os.environ.get("CUTE_DSL_LOG_TO_CONSOLE") is None:
# Need to set log to console, otherwise the log level won't take effect
os.environ["CUTE_DSL_LOG_TO_CONSOLE"] = "1"
# Can also be passed as argument
os.environ["SGLANG_RUN_ID"] = (
f"sglang-run-{time.time()}-{random.randint(0, 100000000)}"