cann-robotrevise kernel compile options: optim

文件	最后提交记录	最后更新时间
examples	补充apply类算子README文档以及example Co-authored-by: chenfeng61<chenfeng61@huawei.com> # message auto-generated for no-merge-commit merge: !3446 merge readme into master 补充apply类算子README文档以及example Created-by: chenfeng61 Commit-by: chenfeng61 Merged-by: cann-robot Description: ## 描述补充apply类算子README文档以及example <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/1869 <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!3446	4 个月前
op_graph	代码同步 Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !450 merge master into master 代码同步 Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!450	7 个月前
op_host	revise kernel compile options: optim Co-authored-by: cong-jiyu<congjiyu@h-partners.com> # message auto-generated for no-merge-commit merge: !5707 merge master into master revise kernel compile options: optim Created-by: cong-jiyu Commit-by: cong-jiyu Merged-by: cann-robot Description: ## 描述 revise kernel compile options: optim ## 关联的Issue [#1985](https://gitcode.com/cann/ops-nn/issues/1985) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!5707	2 个月前
op_kernel	revise kernel compile options: optim Co-authored-by: cong-jiyu<congjiyu@h-partners.com> # message auto-generated for no-merge-commit merge: !5707 merge master into master revise kernel compile options: optim Created-by: cong-jiyu Commit-by: cong-jiyu Merged-by: cann-robot Description: ## 描述 revise kernel compile options: optim ## 关联的Issue [#1985](https://gitcode.com/cann/ops-nn/issues/1985) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!5707	2 个月前
tests	新增 apply_ftrl 算子tiling ut Co-authored-by: cheng-ziyang2<chengziyang2@huawei.com> # message auto-generated for no-merge-commit merge: !2015 merge ftrl_ut into master 新增 apply_ftrl 算子tiling ut Created-by: cheng-ziyang2 Commit-by: cheng-ziyang2 Merged-by: cann-robot Description: ## 描述问题/功能描述本次PR旨在恢复ApplyFtrl算子针对FLOAT32、FLOAT16和BFLOAT16三种数据类型的单元测试，以验证其分片（tiling）逻辑的正确性。这些测试用例先前被注释而无法执行，导致相关功能的测试覆盖不足。通过启用这些测试，可以增强代码质量保障，确保算子在多种数据类型和输入形状下的分片计算符合预期。修改方案描述具体修改是将测试文件中三个被注释的测试函数（apply_ftrl_testcase_float32、apply_ftrl_testcase_float16 和 apply_ftrl_testcase_bfloat16）恢复为可执行状态。同时，更新了每个测试用例中用于验证的预期结果字符串（expectStr），使其与当前实际的分片算法输出保持一致，从而确保测试能够有效运行并通过。 ## 关联的Issue 关联Issue [#1121](https://gitcode.com/cann/ops-nn/issues/1121) ## 测试 ut测试通过 ## 文档更新不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!2015	5 个月前
CMakeLists.txt	代码同步 Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !450 merge master into master 代码同步 Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!450	7 个月前
README.md	修改doc tools低错 Co-authored-by: caiwenwen<caiwenwen6@h-partners.com> # message auto-generated for no-merge-commit merge: !4214 merge master into master 修改doc tools低错 Created-by: caiwenwen Commit-by: caiwenwen Merged-by: cann-robot Description: ## 描述修改nn仓的markdown语法、htlm是否闭合、链接是否可以正常跳转、本地链接是否正常跳转修改低错修改 ## 关联的Issue 关联Issue [#1783](https://gitcode.com/cann/ops-nn/issues/1783) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新全部文档的链接跳转、htlm标签闭合、markdown语法规范问题 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [X] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!4214	3 个月前

ApplyFtrl

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	√
Atlas 训练系列产品	√

功能说明

算子功能：根据Ftrl-proximal方案更新变量"var"。
计算公式：

$accum_new=accum+grad×gradaccum\_new = accum + grad \times grad$

$linear=linear+grad−accum_new−lr_power−accum−lr_powerlr×varlinear = linear + grad - \frac{accum\_new^{-lr\_power} - accum^{-lr\_power}}{lr} \times var$

$quadratic=accum_new−lr_powerlr+2×l2quadratic = \frac{accum\_new^{-lr\_power}}{lr} + 2 \times l2$
- 若 $∣ l i n e a r ∣ > l 1$ :
  $\frac{sign(linear) \times l1 - linear}{quadratic}$
- 否则:
  $v a r = 0.0$
$accum = accum\_new$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
var	输入	待更新的参数张量，应来自Variable。	BFLOAT16、FLOAT16、FLOAT	ND
accum	输入	梯度累积值，应来自Variable。	BFLOAT16、FLOAT16、FLOAT	ND
linear	输入	校正项，应来自Variable。	BFLOAT16、FLOAT16、FLOAT	ND
grad	输入	梯度张量。	BFLOAT16、FLOAT16、FLOAT	ND
lr	输入	缩放因子，标量。	BFLOAT16、FLOAT16、FLOAT	ND
l1	输入	L1正则化系数，标量。	BFLOAT16、FLOAT16、FLOAT	ND
l2	输入	L2正则化系数，标量。	BFLOAT16、FLOAT16、FLOAT	ND
lr_power	输入	缩放因子的幂次，标量。	BFLOAT16、FLOAT16、FLOAT	ND
use_locking	属性	是否使用锁机制保护更新操作，默认为False。仅支持False。	Bool	-
var	输出	更新后的参数张量。	BFLOAT16、FLOAT16、FLOAT	ND

约束说明

输入张量必须具有相同的形状和类型。

调用说明

调用方式	调用样例	说明
图模式调用	test_geir_apply_ftrl	通过算子IR构图方式调用ApplyFtrl算子。