[Bugfix][MoE] Support GELU tanh in FlashInfer B12x MoE (#51819)

Signed-off-by: Andrii Skliar <askliar@nvidia.com>
Co-authored-by: Andrii Skliar <askliar@nvidia.com>
Co-authored-by: OpenAI Codex <codex@openai.com>
This commit is contained in:
Andrii Skliar
2026-08-11 15:07:58 +00:00
committed by GitHub
co-authored by Andrii Skliar OpenAI Codex
parent 87668ab69b
commit 0fb9897df1
@@ -45,6 +45,7 @@ class FlashInferB12xExperts(mk.FusedMoEExpertsModular):
_ACTIVATION_MAP: dict[MoEActivation, str] = {
MoEActivation.SILU: "silu",
MoEActivation.GELU_TANH: "gelu_tanh",
MoEActivation.RELU2_NO_MUL: "relu2",
}
@@ -188,7 +189,11 @@ class FlashInferB12xExperts(mk.FusedMoEExpertsModular):
@staticmethod
def _supports_activation(activation: MoEActivation) -> bool:
return activation in (MoEActivation.SILU, MoEActivation.RELU2_NO_MUL)
return activation in (
MoEActivation.SILU,
MoEActivation.GELU_TANH,
MoEActivation.RELU2_NO_MUL,
)
@staticmethod
def _supports_parallel_config(moe_parallel_config: FusedMoEParallelConfig) -> bool: