diff --git a/evaluation/Qwen3_Qmni_README.md b/evaluation/Qwen3_Qmni_README.md index 659bb53160e5..b56223cb9f7e 100644 --- a/evaluation/Qwen3_Qmni_README.md +++ b/evaluation/Qwen3_Qmni_README.md @@ -57,6 +57,7 @@ echo "launching ${model}" echo "TP=${TP}" # AITER_MOE_PADDING_SIZE=128 \ only for fp8 tp8 + TORCH_NCCL_BLOCKING_WAIT=1 \ AITER_MOE_SMALL_BATCH=1 \ SGLANG_ROCM_USE_AITER_LINEAR_FP8HIPB=0 \ ROCM_QUICK_REDUCE_QUANTIZATION=INT4 \ @@ -77,15 +78,18 @@ --mm-attention-backend "aiter_attn" \ --max-prefill-tokens 32768 \ --disable-radix-cache \ - --page-size 32 \ + --page-size 1024 \ + --kv-cache-dtype fp8_e4m3 \ --mm-enable-dp-encoder \ --enable-aiter-allreduce-fusion \ --max-running-requests 128 \ - --cuda-graph-max-bs 8 + --cuda-graph-max-bs 16 2>&1 | tee log.server.log & ``` - Please add **AITER_MOE_PADDING_SIZE=128** when launch server for fp8 model with tp8. + Please note + - add **AITER_MOE_PADDING_SIZE=128** when launch server for fp8 model with tp8. + - add **TORCH_NCCL_BLOCKING_WAIT=1** as temporal workaround due to a known HIP runtime bug during cuda graph caputure, which that is tracked here: [ROCm/hip#3876](https://github.com/ROCm/hip/issues/3876). # Curl request diff --git a/scripts/ci/sglang_benchmark_workflow.sh b/scripts/ci/sglang_benchmark_workflow.sh index 0767be7932cf..877c88a3977b 100755 --- a/scripts/ci/sglang_benchmark_workflow.sh +++ b/scripts/ci/sglang_benchmark_workflow.sh @@ -28,6 +28,7 @@ if [[ "${TYPE}" == "launch" ]]; then echo "========== LAUNCHING SERVER ========" if [[ "${model_name}" == "Qwen3-VL-235B" ]]; then export SGLANG_USE_AITER=1 + export TORCH_NCCL_BLOCKING_WAIT=1 echo "********** AOT Prebuild aiter kernel start ... **********" cd /aiter python3 op_tests/test_rope.py @@ -86,6 +87,7 @@ if [[ "${TYPE}" == "launch" ]]; then export SGLANG_ROCM_USE_AITER_LINEAR_SHUFFLE=1 export SGLANG_ROCM_USE_AITER_LINEAR_FP8HIPB=0 export ROCM_QUICK_REDUCE_QUANTIZATION=INT4 + export TORCH_NCCL_BLOCKING_WAIT=1 echo "********** AOT Prebuild aiter kernel start ... **********" cd /aiter python3 op_tests/test_rope.py @@ -119,6 +121,7 @@ if [[ "${TYPE}" == "launch" ]]; then export SGLANG_ROCM_USE_AITER_PA_ASM_PRESHUFFLE_LAYOUT=0 export SGLANG_ROCM_USE_AITER_LINEAR_SHUFFLE=1 export SGLANG_ROCM_USE_AITER_LINEAR_FP8HIPB=1 + export TORCH_NCCL_BLOCKING_WAIT=1 python3 -m sglang.launch_server \ --model-path "${model_path}" \ --host localhost \