| Files | Last commit | Last update |
|---|---|---|
feat(segment_sum): 新增算子README.md及geir调用样例 Co-authored-by: zhang-wenbo-beat<zhangwenbo67@huawei.com> # message auto-generated for no-merge-commit merge: !9290 merge segment_sum_example into master feat(segment_sum): 新增算子README.md及geir调用样例 Created-by: zhang-wenbo-beat Commit-by: zhang-wenbo-beat Merged-by: cann-robot Description: ## 描述 本 PR 为 SegmentSum 算子补齐对外文档与可运行调用样例，便于用户理解算子语义并快速上手： - ** index/segment_sum/README.md**：新增算子使用文档，涵盖产品支持情况（仅 Ascend 950PR/950DT 支持）、功能说明（分段求和计算公式与示例）、参数说明（x / segment_ids / y 的数据类型与格式）、约束说明、以及调用方式索引。 - **index/segment_sum/examples/test_geir_segment_sum.cpp**：新增图模式（GEIR）调用样例，通过 ge::Session 构建并运行 SegmentSum 计算图，演示输入 x（shape {4,2}，填充 2.0）与分段索引 segment_ids（{0, 0, 1, 2}）的构图、图编译与执行流程，并将输出落盘为 bin 文件用于验证。 改动原因：SegmentSum 算子此前缺少对外文档与可运行示例，本 PR 补齐这两部分，降低使用门槛。 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/5053 ## 测试 - **样例验证**：样例本身为图模式可运行用例，输入 x = {4,2} 全 2.0、segment_ids = {0, 0, 1, 2}，按分段求和语义输出 y shape 为 {3,2}： - y[0] = x[0] + x[1] = [4, 4] - y[1] = x[2] = [2, 2] - y[2] = x[3] = [2, 2] ## 文档更新 - 新增 index/segment_sum/README.md：SegmentSum 算子说明文档。 - 新增 index/segment_sum/examples/test_geir_segment_sum.cpp：图模式调用样例。 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 - [x] AI辅助编写 See merge request: cann/ops-nn!9290 | 2 months ago |