| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
feat(GemmV3): 2. implement GemmV3BaseKernel. Co-authored-by: void-ptr<baochaojun@huawei.com> Co-authored-by: rueenov11<zhuruyi@huawei.com> # message auto-generated for no-merge-commit merge: !2525 merge 9.0.0 into 9.0.0 feat(GemmV3): 2. implement GemmV3BaseKernel. Created-by: void_ptr Commit-by: rueenov11;void-ptr;void_ptr Merged-by: cann-robot Description: ## 描述 1、新增GemmV3Base kernel，支持alpha、beta!=1输入，将beta* self + alpha*(a@b)的vector计算过程融到mm kernel中，addmm运算使用FP32进行。 2、aclnnAddmmGetWorkspaceSize、aclnnBaddbmmGetWorkspaceSize接口新增cubeMathType=USE_FP32_ADDMM模式，表示在addmm（矩阵乘法与加法融合）运算过程中对FP16/BF16输入进行FP32升精度计算，从而提升特定场景下的数值精度。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!2525 | 6 个月前 | |
blockDim重命名为numBlocks Co-authored-by: chensi888<chensi79@huawei.com> # message auto-generated for no-merge-commit merge: !1350 merge rename_besides_norm into master blockDim重命名为numBlocks Created-by: chensi79 Commit-by: chensi888 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> vector算子blockDim变量重命名为numBlocks ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/861 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 本地测试 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!1350 | 7 个月前 | |
新增kl_div target反向接口 Co-authored-by: xiu_ling_wang<wangxiuling2@h-partners.com> # message auto-generated for no-merge-commit merge: !2478 merge kl_div_target_backward_9.0.0 into 9.0.0 新增kl_div target反向接口 Created-by: xiu_ling_wang Commit-by: xiu_ling_wang Merged-by: cann-robot Description: ## 描述 torch.nn.functional.kl_div算子在模型训练需要target反向，目前只支持self输入的反向。该pr新增一个aclnnKlDivTargetBackward接口，用于支持target反向。 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/1389 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!2478 | 6 个月前 | |
同步工程公共代码 Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !1071 merge master into master 同步工程公共代码 Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述 归一同步工程代码，并更新license ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!1071 | 7 个月前 |