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

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的表示范围时,也能正常处理

ScreenShot_20260224151431.PNG

Special notes for this issue/备注 (Optional / 选填)

likedislike
wei-peng-1991
2月24日 评论:
likedislike
wei-peng-1991
2月24日 评论:

/assign

likedislike
CANN-robotCANN-robot成员
2月24日 将 wei-peng-1991 设为负责人
Wwei-peng-1991
2月24日 issue状态由 进行中 改变为 已完成
Wwei-peng-1991
2月24日 关闭了 issue
CANN-robotCANN-robot成员
2月24日 添加了label:Accepted
CANN-robotCANN-robot成员
3月20日 添加了label:resolved