| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
新增 Bincount 算子AscendC实现贡献 Co-authored-by: ddplys<348478748@qq.com> Co-authored-by: gitcode-bot<noreply@gitcode.com> # message auto-generated for no-merge-commit merge: !3640 merge master into master 【社区任务】新增 Bincount 算子AscendC实现贡献 Created-by: ddplys Commit-by: ddplys;gitcode-bot;update Merged-by: cann-robot Description: ## 描述 新增 **Bincount 算子**（Ascend C 实现，功能对标 PyTorch torch.bincount），贡献至 experimental/math/bincount。 **背景/原因**：ops-math 仓缺少与 torch.bincount 功能一致的 Ascend C 算子；CANN 内置 aclnnBincount 主体计算在 AI_CPU（多 kernel 图，单次约 663~1446us），性能较低。本算子在 AI Core 单 kernel 实现以提升性能。 **功能**：统计一维非负整数张量中各值出现次数（无 weights 计数 / 有 weights 加权累加），输出长度 L = max(max(self)+1, minlength)；仅支持非负整数（与 torch.bincount 一致），含负数运行期检查报错（507015）、不产出结果。 **接口**：两层结构——底层 def 原型 array/size/weights/bins；对外 aclnn self/weights/minlength/out（torch 风格），L2 将 minlength 构造为 size 张量、任意 dtype weights 经 Cast 统一为 float 下发。 **数据类型**：self∈{int8,int16,int32,int64,uint8}；weights∈{float,float16,double,int8,int16,int32,int64,uint8,bool}；out∈{int32,int64,float,double}（含 910B 无原生 fp64 时的 DOUBLE 输出，经 IEEE-754 位拼接产出）。 **关键实现/优化**：① 单遍散射 + 折叠负数检查（删独立 min 预扫描，self 仅搬一遍、同步屏障 2 减 1）；② 单核快路径（小输入 N≤4096 或大输出 L>2048 时单核直算，跳过多核同步/归并固定开销）；③ 多核私有直方图 + 单道 SyncAll 归并；④ 整块 CopyIn + 写出向量化；⑤ 超大输出转 GM 散射回退路径。 ## 关联的Issue 该 PR 关联 Issue：https://gitcode.com/cann/ops-math/issues/2069 ## 测试 - **UT（门禁覆盖）**： - op_host：infershape 3 + tiling 4 = **7 个**，全部 PASSED； - op_api（L2 接口：功能/边界/异常，含空指针、minlength<0、out 非 1 维、空输入、负数运行期校验）：**14 个**，全部 PASSED； - op_kernel（tikicpulib CPU 仿真，计数/加权/各 dtype）：PASSED。 - **真机自验证（Atlas A2 / 910B，AscendOpTest 同机同轮）**：dtype 全矩阵用例精度全部 compare passed；与官方 aclnnBincount 同环境对比，单核优化后本算子 AI Core 单 kernel ~6~22us，加速比中位 ~110×、绝大多数用例远超任务 10X 目标。 - **负数功能**：含负数用例真机验证触发 AI Core 异常、运行时报错 **507015**、算子报错中止。 ## 文档更新 - 更新了 experimental/math/bincount/README.md（算子功能、参数、约束、调用说明）。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3640 | 3 个月前 | |
新增 Bincount 算子AscendC实现贡献 Co-authored-by: ddplys<348478748@qq.com> Co-authored-by: gitcode-bot<noreply@gitcode.com> # message auto-generated for no-merge-commit merge: !3640 merge master into master 【社区任务】新增 Bincount 算子AscendC实现贡献 Created-by: ddplys Commit-by: ddplys;gitcode-bot;update Merged-by: cann-robot Description: ## 描述 新增 **Bincount 算子**（Ascend C 实现，功能对标 PyTorch torch.bincount），贡献至 experimental/math/bincount。 **背景/原因**：ops-math 仓缺少与 torch.bincount 功能一致的 Ascend C 算子；CANN 内置 aclnnBincount 主体计算在 AI_CPU（多 kernel 图，单次约 663~1446us），性能较低。本算子在 AI Core 单 kernel 实现以提升性能。 **功能**：统计一维非负整数张量中各值出现次数（无 weights 计数 / 有 weights 加权累加），输出长度 L = max(max(self)+1, minlength)；仅支持非负整数（与 torch.bincount 一致），含负数运行期检查报错（507015）、不产出结果。 **接口**：两层结构——底层 def 原型 array/size/weights/bins；对外 aclnn self/weights/minlength/out（torch 风格），L2 将 minlength 构造为 size 张量、任意 dtype weights 经 Cast 统一为 float 下发。 **数据类型**：self∈{int8,int16,int32,int64,uint8}；weights∈{float,float16,double,int8,int16,int32,int64,uint8,bool}；out∈{int32,int64,float,double}（含 910B 无原生 fp64 时的 DOUBLE 输出，经 IEEE-754 位拼接产出）。 **关键实现/优化**：① 单遍散射 + 折叠负数检查（删独立 min 预扫描，self 仅搬一遍、同步屏障 2 减 1）；② 单核快路径（小输入 N≤4096 或大输出 L>2048 时单核直算，跳过多核同步/归并固定开销）；③ 多核私有直方图 + 单道 SyncAll 归并；④ 整块 CopyIn + 写出向量化；⑤ 超大输出转 GM 散射回退路径。 ## 关联的Issue 该 PR 关联 Issue：https://gitcode.com/cann/ops-math/issues/2069 ## 测试 - **UT（门禁覆盖）**： - op_host：infershape 3 + tiling 4 = **7 个**，全部 PASSED； - op_api（L2 接口：功能/边界/异常，含空指针、minlength<0、out 非 1 维、空输入、负数运行期校验）：**14 个**，全部 PASSED； - op_kernel（tikicpulib CPU 仿真，计数/加权/各 dtype）：PASSED。 - **真机自验证（Atlas A2 / 910B，AscendOpTest 同机同轮）**：dtype 全矩阵用例精度全部 compare passed；与官方 aclnnBincount 同环境对比，单核优化后本算子 AI Core 单 kernel ~6~22us，加速比中位 ~110×、绝大多数用例远超任务 10X 目标。 - **负数功能**：含负数用例真机验证触发 AI Core 异常、运行时报错 **507015**、算子报错中止。 ## 文档更新 - 更新了 experimental/math/bincount/README.md（算子功能、参数、约束、调用说明）。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3640 | 3 个月前 | |
新增 Bincount 算子AscendC实现贡献 Co-authored-by: ddplys<348478748@qq.com> Co-authored-by: gitcode-bot<noreply@gitcode.com> # message auto-generated for no-merge-commit merge: !3640 merge master into master 【社区任务】新增 Bincount 算子AscendC实现贡献 Created-by: ddplys Commit-by: ddplys;gitcode-bot;update Merged-by: cann-robot Description: ## 描述 新增 **Bincount 算子**（Ascend C 实现，功能对标 PyTorch torch.bincount），贡献至 experimental/math/bincount。 **背景/原因**：ops-math 仓缺少与 torch.bincount 功能一致的 Ascend C 算子；CANN 内置 aclnnBincount 主体计算在 AI_CPU（多 kernel 图，单次约 663~1446us），性能较低。本算子在 AI Core 单 kernel 实现以提升性能。 **功能**：统计一维非负整数张量中各值出现次数（无 weights 计数 / 有 weights 加权累加），输出长度 L = max(max(self)+1, minlength)；仅支持非负整数（与 torch.bincount 一致），含负数运行期检查报错（507015）、不产出结果。 **接口**：两层结构——底层 def 原型 array/size/weights/bins；对外 aclnn self/weights/minlength/out（torch 风格），L2 将 minlength 构造为 size 张量、任意 dtype weights 经 Cast 统一为 float 下发。 **数据类型**：self∈{int8,int16,int32,int64,uint8}；weights∈{float,float16,double,int8,int16,int32,int64,uint8,bool}；out∈{int32,int64,float,double}（含 910B 无原生 fp64 时的 DOUBLE 输出，经 IEEE-754 位拼接产出）。 **关键实现/优化**：① 单遍散射 + 折叠负数检查（删独立 min 预扫描，self 仅搬一遍、同步屏障 2 减 1）；② 单核快路径（小输入 N≤4096 或大输出 L>2048 时单核直算，跳过多核同步/归并固定开销）；③ 多核私有直方图 + 单道 SyncAll 归并；④ 整块 CopyIn + 写出向量化；⑤ 超大输出转 GM 散射回退路径。 ## 关联的Issue 该 PR 关联 Issue：https://gitcode.com/cann/ops-math/issues/2069 ## 测试 - **UT（门禁覆盖）**： - op_host：infershape 3 + tiling 4 = **7 个**，全部 PASSED； - op_api（L2 接口：功能/边界/异常，含空指针、minlength<0、out 非 1 维、空输入、负数运行期校验）：**14 个**，全部 PASSED； - op_kernel（tikicpulib CPU 仿真，计数/加权/各 dtype）：PASSED。 - **真机自验证（Atlas A2 / 910B，AscendOpTest 同机同轮）**：dtype 全矩阵用例精度全部 compare passed；与官方 aclnnBincount 同环境对比，单核优化后本算子 AI Core 单 kernel ~6~22us，加速比中位 ~110×、绝大多数用例远超任务 10X 目标。 - **负数功能**：含负数用例真机验证触发 AI Core 异常、运行时报错 **507015**、算子报错中止。 ## 文档更新 - 更新了 experimental/math/bincount/README.md（算子功能、参数、约束、调用说明）。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3640 | 3 个月前 | |
新增 Bincount 算子AscendC实现贡献 Co-authored-by: ddplys<348478748@qq.com> Co-authored-by: gitcode-bot<noreply@gitcode.com> # message auto-generated for no-merge-commit merge: !3640 merge master into master 【社区任务】新增 Bincount 算子AscendC实现贡献 Created-by: ddplys Commit-by: ddplys;gitcode-bot;update Merged-by: cann-robot Description: ## 描述 新增 **Bincount 算子**（Ascend C 实现，功能对标 PyTorch torch.bincount），贡献至 experimental/math/bincount。 **背景/原因**：ops-math 仓缺少与 torch.bincount 功能一致的 Ascend C 算子；CANN 内置 aclnnBincount 主体计算在 AI_CPU（多 kernel 图，单次约 663~1446us），性能较低。本算子在 AI Core 单 kernel 实现以提升性能。 **功能**：统计一维非负整数张量中各值出现次数（无 weights 计数 / 有 weights 加权累加），输出长度 L = max(max(self)+1, minlength)；仅支持非负整数（与 torch.bincount 一致），含负数运行期检查报错（507015）、不产出结果。 **接口**：两层结构——底层 def 原型 array/size/weights/bins；对外 aclnn self/weights/minlength/out（torch 风格），L2 将 minlength 构造为 size 张量、任意 dtype weights 经 Cast 统一为 float 下发。 **数据类型**：self∈{int8,int16,int32,int64,uint8}；weights∈{float,float16,double,int8,int16,int32,int64,uint8,bool}；out∈{int32,int64,float,double}（含 910B 无原生 fp64 时的 DOUBLE 输出，经 IEEE-754 位拼接产出）。 **关键实现/优化**：① 单遍散射 + 折叠负数检查（删独立 min 预扫描，self 仅搬一遍、同步屏障 2 减 1）；② 单核快路径（小输入 N≤4096 或大输出 L>2048 时单核直算，跳过多核同步/归并固定开销）；③ 多核私有直方图 + 单道 SyncAll 归并；④ 整块 CopyIn + 写出向量化；⑤ 超大输出转 GM 散射回退路径。 ## 关联的Issue 该 PR 关联 Issue：https://gitcode.com/cann/ops-math/issues/2069 ## 测试 - **UT（门禁覆盖）**： - op_host：infershape 3 + tiling 4 = **7 个**，全部 PASSED； - op_api（L2 接口：功能/边界/异常，含空指针、minlength<0、out 非 1 维、空输入、负数运行期校验）：**14 个**，全部 PASSED； - op_kernel（tikicpulib CPU 仿真，计数/加权/各 dtype）：PASSED。 - **真机自验证（Atlas A2 / 910B，AscendOpTest 同机同轮）**：dtype 全矩阵用例精度全部 compare passed；与官方 aclnnBincount 同环境对比，单核优化后本算子 AI Core 单 kernel ~6~22us，加速比中位 ~110×、绝大多数用例远超任务 10X 目标。 - **负数功能**：含负数用例真机验证触发 AI Core 异常、运行时报错 **507015**、算子报错中止。 ## 文档更新 - 更新了 experimental/math/bincount/README.md（算子功能、参数、约束、调用说明）。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3640 | 3 个月前 |
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 3 个月前 | ||
| 3 个月前 | ||
| 3 个月前 | ||
| 3 个月前 |