PR Category

Operator Mechanism

PR Types

Bug fixes

Description

修复bit_count和moe_gate_dispatch_permute问题：
1. bit_count 相关问题:

• 输入x可以是二维情况，paddle infermeta中添加了额外的约束：x只能为1维，导致模型运行错误：

• 输出形状信息推导存在错误，导致后续算子infermeta出错，这里对齐eb中自定义算子shape推导信息：

2. moe_gate_dispatch_permute 问题：

• inferMeta中存在错误约束，当存在可选输入corr_bias时，其shape[0]应对齐专家数即gate_logits.shape[1]，而不是x_dims[0]，因此导致模型报错：

• TODO：模型输出类型并未对齐eb算子（应对齐输入x类型输出bf16，实际输出fp32，导致后续算子infermeta出错），未在infermeta中发现错误，暂时在组网中添加cast绕过：

pcard-67164