core.fusions

Submodules

• core.fusions.fused_weighted_squared_relu
• core.fusions.fused_softmax
• core.fusions.fused_pad_routing_map
• core.fusions.fused_cross_entropy
• core.fusions.fused_bias_dropout
• core.fusions.fused_bias_geglu
• core.fusions.fused_indices_converter
• core.fusions.fused_layer_norm
• core.fusions.fused_bias_swiglu
• core.fusions.fused_bias_gelu
• core.fusions.fused_mla_yarn_rope_apply