Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.
MinimaxSparseAttention算子，A2A5校验dtype不一致，修改修改
A2\A5
A2只校验A2类型，A5只校验A5类型
/assign @renruhai
/assign
Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.
Describe the current behavior / 问题描述 (Mandatory / 必填)
MinimaxSparseAttention算子，A2A5校验dtype不一致，修改修改
Environment / 环境信息 (Mandatory / 必填)
A2\A5
Steps to reproduce the issue / 重现步骤 (Mandatory / 必填)
MinimaxSparseAttention算子，A2A5校验dtype不一致，修改修改
Describe the expected behavior / 预期结果 (Mandatory / 必填)
A2只校验A2类型，A5只校验A5类型
Related log / screenshot / 日志 / 截图 (Mandatory / 必填)
A2只校验A2类型，A5只校验A5类型
Special notes for this issue/备注 (Optional / 选填)