[AMD] Fix v4 topk issue (#37439)

2026-09-01 08:37:30 -07:00
parent 44a92e54b9
commit bb3e3cbceb
1 changed files with 1 additions and 1 deletions
@@ -492,7 +492,7 @@ class DeepseekV4HipRadixBackend(
            page_size=self.page_size,
            page_table=core_attn_metadata.page_table,
            c4_seq_lens=core_attn_metadata.c4_topk_lengths_raw,
-            use_topk_v2=False,
+            use_topk_v2=self.dsa_topk_backend.should_use_topk_v2(),
        )

    def init_forward_metadata_decode(