已合并
add_random_op #30653
Ambi创建于 2月9日
add_random_op #30653
已合并
Ambi创建于 2月9日
已删除 :v2.8.0合入到Ascend/pytorchv2.8.0
Ambi
Ambi
2月9日

【合入来源】

(如有)请关联需求文档/issue链接

【修改方案】

通过对原生rng实现代码打patch的方法注册npu设备的实现。主要主要针对\torch_prims\rng_prims.py 中的philox_rand_offset,philox_rand_offset_meta,register_philox_rand,register_run_and_save_rng_state_op,register_run_with_rng_state_op这些代码。其中_philox_rand 此函数原生调用中不传入设备类型,修改为传入设备npu。

【资料变更】

不涉及

【接口变更】

不涉及

【功能验证】

编写UT进行测试,可以保证在固定随机种子的情况下eager,inductor可以得到相同的结果(如有需要可提供截图)

【CheckList】

PR提交人对以下CheckList自检项进行全量自检,自检通过或不涉及,均修改 [ ] 为 [x]

likedislike
Pull Request已成功合入, 合并人@ascend-robot
(感谢 Ambi 的贡献)
AmbiAmbi
2月9日 创建了 pull request,commit 89e32c8d
ascend-robot
ascend-robot成员
2月9日 评论:

Thanks for your pull-request.
The full list of commands accepted by me can be found at here.
You can get sig-info at here

以下是根据您提交的修改文件推荐的Reviewer和Committer序列,需各模块评审通过后方可合入.

Module List Reviewers Committers
test rain-666, yang-binrong, hbhu_bin, chujinjin, liujunzhu chenhao_1209, shaoyf, guoqi1024, wangqiang160, medivh-x
torch_npu/_inductor liujunzhu, huangyunlong2022, shaoyf, guoqi1024, wangqiang160 zqwenn, chujinjin, wangchao430, wangmin0104, wangqiang160
torch_npu/utils rain-666, weizhan4, shaoyf, wangchao430, htchu suhaibo, li_jing_hw, anyrenwei, guoqi1024, wangmin0104
likedislike
ascend-robotascend-robot成员
2月9日 添加了label:ascend-cla/yes
Ambi
Ambi
2月9日 评论:

compile

likedislike
此处折叠了63条消息 查看更多
ascend-robotascend-robot成员
2月10日 添加了label:approved
htchu成员
2月10日 评论:

/approve

likedislike
ascend-robotascend-robot成员
2月10日 添加了label:lgtm
ascend-robot
ascend-robot成员
2月10日 评论:

Review Guide

This Pull-Request Passes Review.
Committers who wrote a comment of /approve are: zichun_ye, htchu.
Reviewers who wrote a comment of /lgtm are: htchu, zichun_ye.

likedislike
ascend-robotascend-robot成员
2月10日 合入了pull request