Increase post-capture decode memory reserve (#34996)
This commit is contained in:
@@ -4913,7 +4913,7 @@ class ServerArgs:
|
|||||||
if self.post_capture_kv_sizing_planned():
|
if self.post_capture_kv_sizing_planned():
|
||||||
# Post-capture sizing measures free memory after graph capture, so
|
# Post-capture sizing measures free memory after graph capture, so
|
||||||
# skip the graph/activation reserve; keep only the floor + parallel slack.
|
# skip the graph/activation reserve; keep only the floor + parallel slack.
|
||||||
reserved_mem = 512
|
reserved_mem = 1536
|
||||||
reserved_mem += self.tp_size * self.pp_size / 8 * 1024
|
reserved_mem += self.tp_size * self.pp_size / 8 * 1024
|
||||||
else:
|
else:
|
||||||
# Tokens the activation working set scales with (per serving mode).
|
# Tokens the activation working set scales with (per serving mode).
|
||||||
|
|||||||
Reference in New Issue
Block a user