[CI] Run test_unified_radix_cache_kl_dcp on cutedsl_mla with bf16 KV cache (#39553)
This commit is contained in:
@@ -4,7 +4,7 @@ Under DCP the radix layer allocates widened logical indices while each rank's
|
||||
buffers hold only its 1/dcp_size shard, so a missing translation makes cache
|
||||
hits return another rank's KV. The KL cases catch that as a large divergence.
|
||||
|
||||
Blackwell-only: the MLA DCP decode path needs ``tokenspeed_mla`` (SM100/12x).
|
||||
Blackwell-only: ``cutedsl_mla`` is the DCP-native MLA decode kernel on SM100.
|
||||
"""
|
||||
|
||||
import unittest
|
||||
@@ -65,9 +65,7 @@ class TestUnifiedKimiLinearDcpHiCache(UnifiedRadixTreeTestMixin, CustomTestCase)
|
||||
"--page-size",
|
||||
str(PAGE_SIZE),
|
||||
"--attention-backend",
|
||||
"tokenspeed_mla",
|
||||
"--kv-cache-dtype",
|
||||
"fp8_e4m3",
|
||||
"cutedsl_mla",
|
||||
"--dcp-comm-backend",
|
||||
"a2a",
|
||||
"--dcp-replicate-q-proj",
|
||||
|
||||
Reference in New Issue
Block a user