MOE_TOPK_SIGMOID_MAX_TOP_K

Constant MOE_TOPK_SIGMOID_MAX_TOP_K 

Source
pub const MOE_TOPK_SIGMOID_MAX_TOP_K: usize = 32;
Expand description

2026-09-25: Largest top_k the sigmoid routing kernel holds: #define MAX_TOP_K in kernels/gb10/common/moe_topk_sigmoid.cu, which sizes its shared s_top_vals / s_top_idxs. The kernel clamps a larger top_k to it, and NemotronMoeLayer construction refuses one (nemotron_moe.rs). crates/model-engine/tests/moe_topk_sigmoid_bounds.rs pins this mirror to the define and fails if a directory other than common/ holds a copy of the kernel.