Star0

Star0

cann-robot修改 math 类算子的kernel UT 中的error日志误报

6049f22b创建于 2025年10月29日历史提交

文件	最后提交记录	最后更新时间
docs	补齐aclnn列表清单与修改未完全修改aclnn的md文档 Co-authored-by: 付饶<furao15@h-partners.com> # message auto-generated for no-merge-commit merge: merge aclnn_md into master 补齐aclnn列表清单与修改未完全修改aclnn的md文档 Created-by: sakuraqqz Commit-by: 付饶 Merged-by: ascend-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 补齐aclnn列表清单与修改未完全修改aclnn的md文档。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #123--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/6 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!11	11 个月前
examples	update license Co-authored-by: zhaozi3<zhaozi3@h-partners.com> # message auto-generated for no-merge-commit merge: !74 merge master into master update license Created-by: zhaozi3 Commit-by: zhaozi3 Merged-by: ascend-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> This program is free software, you can redistribute it and/or modify.-> This program is free software, you can redistribute it and/or modify it. ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #123--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!74	10 个月前
op_graph	update license Co-authored-by: zhaozi3<zhaozi3@h-partners.com> # message auto-generated for no-merge-commit merge: !74 merge master into master update license Created-by: zhaozi3 Commit-by: zhaozi3 Merged-by: ascend-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> This program is free software, you can redistribute it and/or modify.-> This program is free software, you can redistribute it and/or modify it. ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #123--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!74	10 个月前
op_host	update license Co-authored-by: zhaozi3<zhaozi3@h-partners.com> # message auto-generated for no-merge-commit merge: !74 merge master into master update license Created-by: zhaozi3 Commit-by: zhaozi3 Merged-by: ascend-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> This program is free software, you can redistribute it and/or modify.-> This program is free software, you can redistribute it and/or modify it. ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #123--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!74	10 个月前
op_kernel	update license Co-authored-by: zhaozi3<zhaozi3@h-partners.com> # message auto-generated for no-merge-commit merge: !74 merge master into master update license Created-by: zhaozi3 Commit-by: zhaozi3 Merged-by: ascend-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> This program is free software, you can redistribute it and/or modify.-> This program is free software, you can redistribute it and/or modify it. ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #123--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!74	10 个月前
tests	修改 math 类算子的kernel UT 中的error日志误报 Co-authored-by: 付饶<furao15@h-partners.com> # message auto-generated for no-merge-commit merge: !109 merge master into master 修改 math 类算子的kernel UT 中的error日志误报 Created-by: sakuraqqz Commit-by: 付饶 Merged-by: turing_project1 Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 修改 math 类算子 add_lora，angle_v2， diag_v2，grouped_bias_add_grad，histogram_v2，is_inf，lin_space，non_finite_check，pows，rfft1_d的kernel UT 中的error日志误报，修改结果已附在评论中。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #123--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/76 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!109	10 个月前
CMakeLists.txt	update license Co-authored-by: zhaozi3<zhaozi3@h-partners.com> # message auto-generated for no-merge-commit merge: !74 merge master into master update license Created-by: zhaozi3 Commit-by: zhaozi3 Merged-by: ascend-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> This program is free software, you can redistribute it and/or modify.-> This program is free software, you can redistribute it and/or modify it. ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #123--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!74	10 个月前
README.md	init	11 个月前

AddLora

产品支持情况

产品	是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品	×
Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件	√

功能说明

算子功能：

将输入x根据输入索引indices，分别和对应的weightA，weightB相乘，然后将结果累加到输入y上并输出。
计算公式：

给定输入张量 x，最后一维的长度为 2d，函数 AddLora 进行以下计算：
1. 将x根据indices中的索引进行重排，对应同一组权重的x排列在一起。
2. 循环每个Lora分组，分别拿相应的x和weightA做矩阵乘：
  $x_{i} \cdot weightA[i, layerIdx, :, :]$
3. 得到的Z1继续和weightB做矩阵乘：
  $\cdot weightB[i, layerIdx, :, :] \times scale$
4. 最终把Z2输出累加到y上：
  $out=y[:,yOffset:yOffset+ySliceSize]+Z2\text{out} = y[:, yOffset: yOffset+ySliceSize] + Z2$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
y	输入	公式中的输入y。	FLOAT16	ND
x	输入	公式中的输入x。	FLOAT16	ND
weightB	输入	公式中的输入weightB。	FLOAT16	ND、NZ
indices	输入	公式中的输入indices。	INT32	ND
weightAOptional	输入	公式中的输入weightAOptional。	FLOAT16	ND、NZ
layerIdx	属性	表示层数索引。值需要小于`weightB`的第二个维度`L`。	INT	-
scale	属性	缩放系数。	FLOAT	-
yOffset	属性	y更新时的偏移量。值需要小于`y`的第二个维度`H3`。	INT	-
ySliceSize	属性	y更新时的范围。值需要小于`y`的第二个维度`H3`。	INT	-
out	输出	公式中的out。	FLOAT16	ND

约束说明

无

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_add_lora	通过aclnnAddLora接口方式调用AddLora算子。
图模式调用	test_geir_add_lora	通过算子IR构图方式调用AddLora算子。