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
|
- --served-model-name
|
||||||
- llm
|
- llm
|
||||||
- --max-model-len
|
- --max-model-len
|
||||||
- '256000'
|
- '131072'
|
||||||
- --gpu-memory-utilization
|
- --gpu-memory-utilization
|
||||||
- '0.95'
|
- '0.90'
|
||||||
- --trust-remote-code
|
- --trust-remote-code
|
||||||
- -tp
|
- -tp
|
||||||
- '4'
|
- '4'
|
||||||
- --max-num-seqs
|
- --max-num-seqs
|
||||||
- '2'
|
- '1'
|
||||||
- --disable-log-requests
|
- --disable-log-requests
|
||||||
- --disable-frontend-multiprocessing
|
- --disable-frontend-multiprocessing
|
||||||
- --max-num-batched-tokens
|
- --max-num-batched-tokens
|
||||||
- '4096'
|
- '8192'
|
||||||
- --enable-chunked-prefill
|
- --enable-chunked-prefill
|
||||||
- --max-seq-len-to-capture
|
- --max-seq-len-to-capture
|
||||||
- '32768'
|
- '32768'
|
||||||
@@ -45,3 +45,5 @@ env:
|
|||||||
value: admission64
|
value: admission64
|
||||||
- name: BI100_GDN_RESTORE_MODE
|
- name: BI100_GDN_RESTORE_MODE
|
||||||
value: hybrid64
|
value: hybrid64
|
||||||
|
- name: BI100_MOE_COREX_TOPK_SOFTMAX
|
||||||
|
value: '0'
|
||||||
|
|||||||
Reference in New Issue
Block a user