Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.
自动融合CV融合要融合where、compare这类算子,它的数据位宽会比matmul输出类型小,这个时候我们的stage算的时候会导致vector UB数据计算无法统一
Ascend 950 支持cv融合
自动融合,CV融合 matmul+ where、compare这类算子融合
计算正确,融合后与单算子比精度达到预期
现在我们要融合where、compare这类算子,它的数据位宽会比matmul输出类型小,这个时候我们的stage算的时候会导致vector UB数据计算无法统一,现在我们把每次的stage统一到处理矩阵的行
Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.
Describe the current behavior / 问题描述 (Mandatory / 必填)
自动融合CV融合要融合where、compare这类算子,它的数据位宽会比matmul输出类型小,这个时候我们的stage算的时候会导致vector UB数据计算无法统一
Environment Information / 环境信息 (Mandatory / 必填)
Ascend 950
支持cv融合
Steps to reproduce the issue / 重现步骤 (Mandatory / 必填)
自动融合,CV融合 matmul+ where、compare这类算子融合
Describe the expected behavior / 预期结果 (Mandatory / 必填)
计算正确,融合后与单算子比精度达到预期
Related log / screenshot / 日志 / 截图 (Mandatory / 必填)
现在我们要融合where、compare这类算子,它的数据位宽会比matmul输出类型小,这个时候我们的stage算的时候会导致vector UB数据计算无法统一,现在我们把每次的stage统一到处理矩阵的行
Special notes for this issue/备注 (Optional / 选填)