已关闭
[Bug-Report|缺陷反馈]: 修复NLLLossGrad算子代码中ignore_index参数的数据类型不一致问题,统一改为int64_t #1105
wei-peng-1991创建于 2月24日关闭于 2月24日
wei-peng-1991
2月24日 评论:
2月24日 评论:
/assgin @wei-peng-1991


wei-peng-1991
2月24日 评论:
2月24日 评论:
/assign


2月24日 将 wei-peng-1991 设为负责人
2月24日 issue状态由 进行中 改变为 已完成
2月24日 关闭了 issue
2月24日 添加了label:Accepted
3月20日 添加了label:resolved
Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.
Describe the current behavior / 问题描述 (Mandatory / 必填)
修复NLLLossGrad算子及其核函数中ignore_index参数的数据类型不一致问题。该参数在算子定义、Tiling数据结构、核函数参数及日志打印等多个环节存在int32_t与int64_t的混用,可能导致在处理较大索引值(如大规模分类任务或特定负值)时发生数据截断、整数溢出或类型转换错误,从而影响算子计算的正确性和鲁棒性。
Environment / 环境信息 (Mandatory / 必填)
Ascend950
Steps to reproduce the issue / 重现步骤 (Mandatory / 必填)
NllLossGrad算子中属中ignore_index的值超过了int32的表示范围
Describe the expected behavior / 预期结果 (Mandatory / 必填)
NllLossGrad算子中属性ignore_index的值超过int32的表示范围时,也能正常处理
Related log / screenshot / 日志 / 截图 (Mandatory / 必填)
Special notes for this issue/备注 (Optional / 选填)