0
代码
代码
Issues
Pull Requests
讨论
Wiki
分析
0
cann-robotcann-robot新增CoalesceSparse算子A5实现（SIMT）
d88f4b67创建于 20 小时前历史提交
文件最后提交记录最后更新时间
docs
新增CoalesceSparse算子A5实现（SIMT） Co-authored-by: l00855153<liudingming3@huawei.com> # message auto-generated for no-merge-commit merge: !2975 merge master into master 新增CoalesceSparse算子A5实现（SIMT） Created-by: Tower19 Commit-by: l00855153;Tower19 Merged-by: cann-robot Description: ## 描述 新增CoalesceSparse算子A5实现（SIMT+AtomicAdd） ## 测试 自验证通过（60 case） ## 文档更新 更新ops-math/conversion/coalesce_sparse/docs/aclnnCoalesceSparse.md ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!297520 小时前
examples
新增CoalesceSparse算子A5实现（SIMT） Co-authored-by: l00855153<liudingming3@huawei.com> # message auto-generated for no-merge-commit merge: !2975 merge master into master 新增CoalesceSparse算子A5实现（SIMT） Created-by: Tower19 Commit-by: l00855153;Tower19 Merged-by: cann-robot Description: ## 描述 新增CoalesceSparse算子A5实现（SIMT+AtomicAdd） ## 测试 自验证通过（60 case） ## 文档更新 更新ops-math/conversion/coalesce_sparse/docs/aclnnCoalesceSparse.md ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!297520 小时前
op_graph
增加A2 A3新算子支持 Co-authored-by: slx2008<shaolixin@huawei.com> # message auto-generated for no-merge-commit merge: !448 merge master into master 增加A2 A3新算子支持 Created-by: songkai111 Commit-by: slx2008 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4485 个月前
op_host
新增CoalesceSparse算子A5实现（SIMT） Co-authored-by: l00855153<liudingming3@huawei.com> # message auto-generated for no-merge-commit merge: !2975 merge master into master 新增CoalesceSparse算子A5实现（SIMT） Created-by: Tower19 Commit-by: l00855153;Tower19 Merged-by: cann-robot Description: ## 描述 新增CoalesceSparse算子A5实现（SIMT+AtomicAdd） ## 测试 自验证通过（60 case） ## 文档更新 更新ops-math/conversion/coalesce_sparse/docs/aclnnCoalesceSparse.md ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!297520 小时前
op_kernel
新增CoalesceSparse算子A5实现（SIMT） Co-authored-by: l00855153<liudingming3@huawei.com> # message auto-generated for no-merge-commit merge: !2975 merge master into master 新增CoalesceSparse算子A5实现（SIMT） Created-by: Tower19 Commit-by: l00855153;Tower19 Merged-by: cann-robot Description: ## 描述 新增CoalesceSparse算子A5实现（SIMT+AtomicAdd） ## 测试 自验证通过（60 case） ## 文档更新 更新ops-math/conversion/coalesce_sparse/docs/aclnnCoalesceSparse.md ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!297520 小时前
tests
新增CoalesceSparse算子A5实现（SIMT） Co-authored-by: l00855153<liudingming3@huawei.com> # message auto-generated for no-merge-commit merge: !2975 merge master into master 新增CoalesceSparse算子A5实现（SIMT） Created-by: Tower19 Commit-by: l00855153;Tower19 Merged-by: cann-robot Description: ## 描述 新增CoalesceSparse算子A5实现（SIMT+AtomicAdd） ## 测试 自验证通过（60 case） ## 文档更新 更新ops-math/conversion/coalesce_sparse/docs/aclnnCoalesceSparse.md ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!297520 小时前
CMakeLists.txt
增加A2 A3新算子支持 Co-authored-by: slx2008<shaolixin@huawei.com> # message auto-generated for no-merge-commit merge: !448 merge master into master 增加A2 A3新算子支持 Created-by: songkai111 Commit-by: slx2008 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4485 个月前
README.md
新增CoalesceSparse算子A5实现（SIMT） Co-authored-by: l00855153<liudingming3@huawei.com> # message auto-generated for no-merge-commit merge: !2975 merge master into master 新增CoalesceSparse算子A5实现（SIMT） Created-by: Tower19 Commit-by: l00855153;Tower19 Merged-by: cann-robot Description: ## 描述 新增CoalesceSparse算子A5实现（SIMT+AtomicAdd） ## 测试 自验证通过（60 case） ## 文档更新 更新ops-math/conversion/coalesce_sparse/docs/aclnnCoalesceSparse.md ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!297520 小时前
README.md

CoalesceSparse

产品支持情况

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

功能说明

  • 算子功能：将相同坐标点的value进行累加求和，进而减少Coo_Tensor的内存大小。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
unique_len 输入 去重后的索引数。 INT32、INT64 ND
unique_indices 输入 去重后的索引数组。 INT32、INT64 ND
indices 输入 索引数组。 INT32、INT64 ND
values 输入 每个坐标对应的元素值。 INT32、FLOAT16、FLOAT32 ND
new_indices 输出 合并后的索引数组。 INT32、INT64 ND
new_values 输出 合并后的元素值。 INT32、FLOAT16、FLOAT32 ND

约束说明

重索引后的indices值不能超过int32上限。

调用说明

调用方式 样例代码 说明
图模式调用 test_aclnn_coalesce_sparse 通过aclnnCoalesceSparse接口方式调用CoalesceSparse算子。