已合并
[bugfix] modify the yaml configuration of op kl_div. #2104
weixin_40805272创建于 2025年1月23日
[bugfix] modify the yaml configuration of op kl_div. #2104
已合并
weixin_40805272创建于 2025年1月23日
refs/pull/2104/head合入到master
weixin_40805272
weixin_40805272
2025年1月23日
  1. [bugfix] modify the yaml configuration of op kl_div.

What type of PR is this?

Uncomment only one /kind <> line, hit enter to put that in a new line, and remove leading whitespaces from that line:

/kind bug
/kind task
/kind feature

What does this PR do / why do we need it:

算子yaml配置有问题,一个入参设置可导,计算会被拦截,cuda不会被拦截,所以更改yaml配置,保持跟cuda一致。
Special notes for your reviewers:

likedislike
Pull Request已成功合入, 合并人@
(感谢 weixin_40805272 的贡献)
weixin_40805272
weixin_40805272
2025年1月23日 评论:

compile

likedislike
atlasaccount1
atlasaccount1
2025年1月23日 评论:

ut failed

likedislike
weixin_40805272
weixin_40805272
2025年1月23日 评论:

compile

likedislike
atlasaccount1
atlasaccount1
2025年1月23日 评论:

ut failed

likedislike
weixin_40805272
weixin_40805272
2025年1月24日 评论:

compile

likedislike
此处折叠了111条消息 查看更多
atlasaccount1
atlasaccount1
2025年2月6日 评论:

流水线任务触发成功,正在执行,请稍候。 任务链接[c05484941d214107b26346bb742f44f2]

likedislike
atlasaccount1
atlasaccount1
2025年2月6日 评论:

checklist流水线任务已触发,正在执行,请稍候。任务链接[56604c7f552747d0997b7b2a5433f4d2]

likedislike
atlasaccount1
atlasaccount1
2025年2月6日 评论:
检查项 状态 详情
code_check >>>
sca >>>
anti_poison >>>
Build_master_x86 >>>
UT_master_x86 >>>
Build_v1110_x86 >>>
UT_v1110_x86 >>>
Build_v210_x86 >>>
UT_v210_x86 >>>
Build_v220_x86 >>>
UT_V220_x86 >>>
Build_v231_x86 >>>
UT_V231_x86 >>>
Build_v240_x86 >>>
UT_v240_x86 >>>
Build_v251_x86 >>>
UT_V251_x86 >>>
judge_master >>>
judge_master >>>
Build_ARM_USE_CXX_ABI >>>
Build_v1110_ARM >>>
Build_v210_ARM >>>
UT_v210_ARM_A2 >>>
Build_v220_ARM >>>
Build_v231_ARM >>>
Build_v240_ARM >>>
Build_v251_ARM >>>
likedislike
wang-guangbin
2025年2月6日 评论:

/lgtm

likedislike
zqwen2025年2月12日进行代码检视1
test/test_base_ops/test_kl_div.py
@@ -101,1 +101,4 @@
101101 
102+ def test_kl_div_input_requires_grad_True(self):
103+ input_cpu = torch.tensor([0.2, 0.7, 0.1], requires_grad=True)
104+ target_cpu = torch.tensor([0., 1., 0.])
zqwen2025年2月12日评论:

修改的是target的yaml,配套用例应该设置target的requires_grad=True

likedislike