| 修改stateless_drop_out_gen_mask算子+aclnn接口
Co-authored-by: huairuchen<chenhuairu@huawei.com>
# message auto-generated for no-merge-commit merge:
!2299 merge pr_dropout into master
修改stateless_drop_out_gen_mask算子+aclnn接口
Created-by: huairuchen
Commit-by: huairuchen
Merged-by: cann-robot
Description: ## 描述
<!--在这里详细描述你的改动,包括改动的原因和所采取的方法。-->
修改aclnnDropoutGenMaskV2Tensor接口,使这个接口可以在A5芯片下正确调用stateless_drop_out_gen_mask算子。
并且对应修改stateless_drop_out_gen_mask算子,使其可以接受aclnnDropoutGenMaskV2Tensor接口的调用。
保证精度正常,性能不劣化
## 关联的Issue
<!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。-->
<!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。-->
https://gitcode.com/cann/ops-math/issues/1401
## 测试
<!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。-->
进行了静态检查 二级冒烟 构建触发
UT测试
泛化ST测试
跑了ttk xrun测试
## 文档更新
<!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。-->
更新了aclnnDropoutGenMaskV2Tensor.md 现接口支持950系产品。
## 类型标签
<!-- [x] 表示选中 -->
- [ ] Bug修复
- [x] 新特性
- [ ] 性能优化
- [ ] 文档更新
- [ ] 其他,请描述:
See merge request: cann/ops-math!2299 | 1 个月前 |