fix(stability): prevent OOM crash + disable garbled topk_softmax kernel
Sub 655 analysis: 634/881 connection errors (server crash during replay). Root cause: max-model-len=256000 + gpu-memory-utilization=0.95 + max-num-seqs=2 caused OOM on long-context requests (128K+ tokens). Changes: - max-model-len: 256000 → 131072 (enough for replay, prevents OOM) - gpu-memory-utilization: 0.95 → 0.90 (safety margin) - max-num-seqs: 2 → 1 (avoid concurrent long-context OOM) - max-num-batched-tokens: 4096 → 8192 (match proven config) - BI100_MOE_COREX_TOPK_SOFTMAX=0 (CUB kernel causes garbled output on BI-V100; PyTorch topk+softmax path is correct and fast enough) Expected impact: server stays alive through entire replay+opencompass run. Sub 655 successful requests had output_tps_avg=11.5 — the TPS is fine, we just need the server to not crash.
This commit is contained in:
@@ -8,18 +8,18 @@ command:
|
||||
- --served-model-name
|
||||
- llm
|
||||
- --max-model-len
|
||||
- '256000'
|
||||
- '131072'
|
||||
- --gpu-memory-utilization
|
||||
- '0.95'
|
||||
- '0.90'
|
||||
- --trust-remote-code
|
||||
- -tp
|
||||
- '4'
|
||||
- --max-num-seqs
|
||||
- '2'
|
||||
- '1'
|
||||
- --disable-log-requests
|
||||
- --disable-frontend-multiprocessing
|
||||
- --max-num-batched-tokens
|
||||
- '4096'
|
||||
- '8192'
|
||||
- --enable-chunked-prefill
|
||||
- --max-seq-len-to-capture
|
||||
- '32768'
|
||||
@@ -45,3 +45,5 @@ env:
|
||||
value: admission64
|
||||
- name: BI100_GDN_RESTORE_MODE
|
||||
value: hybrid64
|
||||
- name: BI100_MOE_COREX_TOPK_SOFTMAX
|
||||
value: '0'
|
||||
|
||||
Reference in New Issue
Block a user