[QUESTION] Can fp8_paged_mqa_logits support next_n > 2?

[starwang1024]

I noticed that the code here in fp8_paged_mqa_logits only supports next_n <= 2. May I ask what the reason for this is? Additionally, what modifications would be needed to support next_n > 2?
Any reply would be very helpful.

[LyricZhao]

Internally at DeepSeek, we don't use next_n > 2. But the NV https://github.com/deepseek-ai/DeepGEMM/tree/nv_dev branch supports such features.

[Paiiiiiiiiiiiiii]

Internally at DeepSeek, we don't use next_n > 2. But the NV https://github.com/deepseek-ai/DeepGEMM/tree/nv_dev branch supports such features.

I noticed that there is one line (DG_HOST_ASSERT(next_n == 1 or next_n == 2 or next_n == 4);), So now it still dont support nextn=3?