已合并
add_random_op #30653
Ambi创建于 2月9日
add_random_op #30653
已合并
从已删除 :v2.8.0合入到Ascend/pytorchv2.8.0
Pull Request已成功合入, 合并人@ascend-robot
(感谢 Ambi 的贡献)AtlasAccount
2月9日 评论:
2月9日 评论:
ascend-robot
2月9日 评论:
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 |


2月9日 添加了label:ascend-cla/yes
Ambi
2月9日 评论:
2月9日 评论:
compile


此处折叠了63条消息 查看更多
2月10日 添加了label:approved
2月10日 添加了label:lgtm
ascend-robot
2月10日 评论:
2月10日 评论:
2月10日 合入了pull request
【合入来源】
【修改方案】
通过对原生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】