Fix CP in-seq-split method for DeepSeek V32 and update related tests (#21192)
This commit is contained in:
@@ -1530,7 +1530,7 @@ class ServerArgs:
|
||||
assert (
|
||||
self.tp_size == 8
|
||||
), "Current multi-machine CP support suffers from precision issues. So context parallel only support Single machine(tp_size == 8)"
|
||||
self.attn_cp_size = self.tp_size
|
||||
self.attn_cp_size = self.tp_size // self.dp_size
|
||||
|
||||
logger.warning(
|
||||
f"Enable Context Parallel opt for deeeseekv3.2-DSA, Setting dp_size == {self.dp_size} and moe_dense_tp_size == {self.moe_dense_tp_size}, ep_size == {self.ep_size}, tp_size == {self.tp_size}, kv_cache_dtype == {self.kv_cache_dtype}, moe_a2a_backend {self.moe_a2a_backend} "
|
||||
|
||||
Reference in New Issue
Block a user