| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
sparse_segment_mean和concat_offset算子支持Ascend950 Co-authored-by: 张鑫<zhangxin660@h-partners.com> # message auto-generated for no-merge-commit merge: !2328 merge move_nn_dst into master sparse_segment_mean和concat_offset算子支持Ascend950 Created-by: z30075199 Commit-by: 张鑫 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> **sparse_segment_mean和concat_offset算子支持Ascend950** concat_offset：计算concat的输入在其输出中的偏移量 <table style="undefined;table-layout: fixed; width: 1479px"><colgroup> <col style="width: 163px"> <col style="width: 180px"> <col style="width: 240px"> <col style="width: 200px"> <col style="width: 197px"> <col style="width: 114px"> <col style="width: 156px"> <col style="width: 145px"> </colgroup> <thead> <tr> <th>参数名</th> <th>输入/输出/属性</th> <th>描述</th> <th>数据类型</th> <th>数据格式</th> </tr></thead> <tbody> <tr> <td>concat_dim</td> <td>输入</td> <td>输入aclTensor</td> <td>INT32</td> <td>ND</td> </tr> <tr> <td>x</td> <td>输入</td> <td>输入TensorList</td> <td>INT32</td> <td>ND</td> </tr> <tr> <td>N</td> <td>属性</td> <td>输入Tensor的个数</td> <td>INT</td> <td>-</td> </tr> <tr> <td>y</td> <td>输出</td> <td>输出TensorList</td> <td>INT32</td> <td>ND</td> </tr> </tbody></table> sparse_segment_mean：计算张量稀疏片段的均值 <table style="undefined;table-layout: fixed; width: 1479px"><colgroup> <col style="width: 163px"> <col style="width: 180px"> <col style="width: 240px"> <col style="width: 200px"> <col style="width: 197px"> <col style="width: 114px"> <col style="width: 156px"> <col style="width: 145px"> </colgroup> <thead> <tr> <th>参数名</th> <th>输入/输出</th> <th>描述</th> <th>数据类型</th> <th>数据格式</th> </tr></thead> <tbody> <tr> <td>x</td> <td>输入</td> <td>输入aclTensor</td> <td>FLOAT, DOUBLE, FLOAT16, BFLOAT16</td> <td>ND</td> </tr> <tr> <td>indices</td> <td>输入</td> <td>输入aclTensor</td> <td>INT32, INT64</td> <td>ND</td> </tr> <tr> <td>segment_ids</td> <td>输入</td> <td>输入aclTensor</td> <td>INT32, INT64</td> <td>ND</td> </tr> <tr> <td>y</td> <td>输出</td> <td>输出aclTensor</td> <td>FLOAT, DOUBLE, FLOAT16, BFLOAT16</td> <td>ND</td> </tr> </tbody></table> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/1558 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!2328 | 5 个月前 | |
sparse_segment_mean分核策略优化 Co-authored-by: 张鑫<zhangxin660@h-partners.com> # message auto-generated for no-merge-commit merge: !7050 merge ssm910 into 9.1.0 sparse_segment_mean分核策略优化 Created-by: z30075199 Commit-by: 张鑫 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 优化sparse_segment_mean分核策略 row上的核尽可能少(每个核处理尽可能多的索引)，同时，col上的每个核处理的数据块根据segment_id重复度不能小于min(4k/8K/9K, innerSize) ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/3889 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 泛化用例500条，平均性能提升25% ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [x] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!7050 | 1 个月前 | |
算子的simt的cpp风格转换为c Co-authored-by: xiaodong666<yuanxiaodong6@huawei.com> # message auto-generated for no-merge-commit merge: !4589 merge to_simt_c into master 算子的simt的cpp风格转换为c Created-by: xiaodong666 Commit-by: xiaodong666 Merged-by: cann-robot Description: ## 描述 simt-c风格转换 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> 关联Issue [#2687](https://gitcode.com/cann/ops-nn/issues/2687) ## 测试 本地修改前后md5sum一致 ## 文档更新 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!4589 | 3 个月前 | |
sparse_segment_mean和concat_offset算子支持Ascend950 Co-authored-by: 张鑫<zhangxin660@h-partners.com> # message auto-generated for no-merge-commit merge: !2328 merge move_nn_dst into master sparse_segment_mean和concat_offset算子支持Ascend950 Created-by: z30075199 Commit-by: 张鑫 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> **sparse_segment_mean和concat_offset算子支持Ascend950** concat_offset：计算concat的输入在其输出中的偏移量 <table style="undefined;table-layout: fixed; width: 1479px"><colgroup> <col style="width: 163px"> <col style="width: 180px"> <col style="width: 240px"> <col style="width: 200px"> <col style="width: 197px"> <col style="width: 114px"> <col style="width: 156px"> <col style="width: 145px"> </colgroup> <thead> <tr> <th>参数名</th> <th>输入/输出/属性</th> <th>描述</th> <th>数据类型</th> <th>数据格式</th> </tr></thead> <tbody> <tr> <td>concat_dim</td> <td>输入</td> <td>输入aclTensor</td> <td>INT32</td> <td>ND</td> </tr> <tr> <td>x</td> <td>输入</td> <td>输入TensorList</td> <td>INT32</td> <td>ND</td> </tr> <tr> <td>N</td> <td>属性</td> <td>输入Tensor的个数</td> <td>INT</td> <td>-</td> </tr> <tr> <td>y</td> <td>输出</td> <td>输出TensorList</td> <td>INT32</td> <td>ND</td> </tr> </tbody></table> sparse_segment_mean：计算张量稀疏片段的均值 <table style="undefined;table-layout: fixed; width: 1479px"><colgroup> <col style="width: 163px"> <col style="width: 180px"> <col style="width: 240px"> <col style="width: 200px"> <col style="width: 197px"> <col style="width: 114px"> <col style="width: 156px"> <col style="width: 145px"> </colgroup> <thead> <tr> <th>参数名</th> <th>输入/输出</th> <th>描述</th> <th>数据类型</th> <th>数据格式</th> </tr></thead> <tbody> <tr> <td>x</td> <td>输入</td> <td>输入aclTensor</td> <td>FLOAT, DOUBLE, FLOAT16, BFLOAT16</td> <td>ND</td> </tr> <tr> <td>indices</td> <td>输入</td> <td>输入aclTensor</td> <td>INT32, INT64</td> <td>ND</td> </tr> <tr> <td>segment_ids</td> <td>输入</td> <td>输入aclTensor</td> <td>INT32, INT64</td> <td>ND</td> </tr> <tr> <td>y</td> <td>输出</td> <td>输出aclTensor</td> <td>FLOAT, DOUBLE, FLOAT16, BFLOAT16</td> <td>ND</td> </tr> </tbody></table> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/1558 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!2328 | 5 个月前 | |
sparse_segment_mean和concat_offset算子支持Ascend950 Co-authored-by: 张鑫<zhangxin660@h-partners.com> # message auto-generated for no-merge-commit merge: !2328 merge move_nn_dst into master sparse_segment_mean和concat_offset算子支持Ascend950 Created-by: z30075199 Commit-by: 张鑫 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> **sparse_segment_mean和concat_offset算子支持Ascend950** concat_offset：计算concat的输入在其输出中的偏移量 <table style="undefined;table-layout: fixed; width: 1479px"><colgroup> <col style="width: 163px"> <col style="width: 180px"> <col style="width: 240px"> <col style="width: 200px"> <col style="width: 197px"> <col style="width: 114px"> <col style="width: 156px"> <col style="width: 145px"> </colgroup> <thead> <tr> <th>参数名</th> <th>输入/输出/属性</th> <th>描述</th> <th>数据类型</th> <th>数据格式</th> </tr></thead> <tbody> <tr> <td>concat_dim</td> <td>输入</td> <td>输入aclTensor</td> <td>INT32</td> <td>ND</td> </tr> <tr> <td>x</td> <td>输入</td> <td>输入TensorList</td> <td>INT32</td> <td>ND</td> </tr> <tr> <td>N</td> <td>属性</td> <td>输入Tensor的个数</td> <td>INT</td> <td>-</td> </tr> <tr> <td>y</td> <td>输出</td> <td>输出TensorList</td> <td>INT32</td> <td>ND</td> </tr> </tbody></table> sparse_segment_mean：计算张量稀疏片段的均值 <table style="undefined;table-layout: fixed; width: 1479px"><colgroup> <col style="width: 163px"> <col style="width: 180px"> <col style="width: 240px"> <col style="width: 200px"> <col style="width: 197px"> <col style="width: 114px"> <col style="width: 156px"> <col style="width: 145px"> </colgroup> <thead> <tr> <th>参数名</th> <th>输入/输出</th> <th>描述</th> <th>数据类型</th> <th>数据格式</th> </tr></thead> <tbody> <tr> <td>x</td> <td>输入</td> <td>输入aclTensor</td> <td>FLOAT, DOUBLE, FLOAT16, BFLOAT16</td> <td>ND</td> </tr> <tr> <td>indices</td> <td>输入</td> <td>输入aclTensor</td> <td>INT32, INT64</td> <td>ND</td> </tr> <tr> <td>segment_ids</td> <td>输入</td> <td>输入aclTensor</td> <td>INT32, INT64</td> <td>ND</td> </tr> <tr> <td>y</td> <td>输出</td> <td>输出aclTensor</td> <td>FLOAT, DOUBLE, FLOAT16, BFLOAT16</td> <td>ND</td> </tr> </tbody></table> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/1558 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!2328 | 5 个月前 |