fix: cuda-graph-max-bs=64 (256 OOMs during capture)
This commit is contained in:
parent
cd362c5ca0
commit
6d3338244b
@ -35,7 +35,7 @@ NCCL_DEBUG="${NCCL_DEBUG:-WARN}"
|
|||||||
# max_running_requests must >= max concurrency tested (128) to avoid queueing
|
# max_running_requests must >= max concurrency tested (128) to avoid queueing
|
||||||
# cuda_graph_max_bs_decode must >= max concurrent batch size for accurate decode perf
|
# cuda_graph_max_bs_decode must >= max concurrent batch size for accurate decode perf
|
||||||
MEM_FRACTION_STATIC="${MEM_FRACTION_STATIC:-0.9}"
|
MEM_FRACTION_STATIC="${MEM_FRACTION_STATIC:-0.9}"
|
||||||
CUDA_GRAPH_MAX_BS_DECODE="${CUDA_GRAPH_MAX_BS_DECODE:-256}"
|
CUDA_GRAPH_MAX_BS_DECODE="${CUDA_GRAPH_MAX_BS_DECODE:-64}"
|
||||||
MAX_RUNNING_REQUESTS="${MAX_RUNNING_REQUESTS:-256}"
|
MAX_RUNNING_REQUESTS="${MAX_RUNNING_REQUESTS:-256}"
|
||||||
|
|
||||||
# ---- Benchmark ----
|
# ---- Benchmark ----
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user