Star0

Star0

cann-robot算子资料修改后同步修改example

dc6d8a58创建于 16 天前历史提交

文件	最后提交记录	最后更新时间
docs	数学类算子资料修改 Co-authored-by: FelixTang7<tangyanfeng@huawei.com> # message auto-generated for no-merge-commit merge: !5432 merge md into master 数学类算子资料修改 Created-by: FelixTang7 Commit-by: FelixTang7 Merged-by: cann-robot Description: ## 描述部分数学类算子资料不清晰，需要进行修改补齐 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/3040 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5432	19 天前
examples	算子资料修改后同步修改example Co-authored-by: FelixTang7<tangyanfeng@huawei.com> # message auto-generated for no-merge-commit merge: !5542 merge md into master 算子资料修改后同步修改example Created-by: FelixTang7 Commit-by: FelixTang7 Merged-by: cann-robot Description: ## 描述部分算子资料有误，修改算子文档后同步对example进行修改 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/3040 ## 测试文档修改，无需测试 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5542	16 天前
framework	refactor: npu_indexing/correlation/one_hot_d onnx 插件迁移至 ParseParamsByOperatorFn 接口 Co-authored-by: clinglai0517<laichangling@huawei.com> # message auto-generated for no-merge-commit merge: !5325 merge fix-onnx-partitionedcall into master refactor: npu_indexing/correlation/one_hot_d onnx 插件迁移至 ParseParamsByOperatorFn 接口 Created-by: clinglai0517 Commit-by: clinglai0517 Merged-by: cann-robot Description: ## 描述将 npu_indexing、correlation、one_hot_d 三个 ONNX 插件从旧 ParseParamsFn(const Message, ge::Operator&) 接口迁移至 ParseParamsByOperatorFn，去除onnx_common.h / ge::onnx:: / NodeProto 依赖（强依赖 ge_onnx.pb.h），插件与 framework UT 编译均脱离 protobuf。 ### 迁移内容 - conversion/strided_slice/framework/npu_indexing_onnx_plugin.cpp + 7 UT ：begins/ends/strides 与五个 mask 经 json 解析。 - common/src/framework/correlation_onnx_plugin.cpp + 5 UT ：groups 属性经 json 解析，保留默认值。 - math/one_hot/framework/one_hot_d_onnx_plugin.cpp + 7 UT ：depth（必需）/ num_classes / on_value / off_value 经 json 解析。 ### 公共工具下沉 onnx_common.h 中所有无 protobuf 依赖的 helper 全部下沉至common/inc/framework/math_onnx_plugin_util.h（protobuf-free）：Vec2Tensor、CreateScalar、DataTypeOnnx、onnx2om_dtype_map、GetOmDtypeFromOnnxDtype、ChangeFormatFromOnnx。onnx_common.h 现仅保留ge_onnx.pb.h 等 include 链供未迁移插件使用，经包含 util 仍提供这些 helper，无破坏。 ### CI 修复 1. scripts/ci/gen_ci_cmd.py：get_op_ut_types 识别 <op>/framework/ 与 <op>/tests/ut/framework/ 变更为 ophost，修复仅改 framework/onnx 插件的 PR 生成 0 条命令、UT 未运行的问题。 2. cmake/ut.cmake：framework 分支 --ops 过滤区分 common 集中目录（按文件名）与独立算子目录（按目录算子名），修复 test__onnx_plugin.cpp 因文件名算子名与目录算子名不一致被 continue 跳过、framework_ut_cases 仅剩 empty.cpp 占位的问题。 ## 关联的Issue Issue #2892 ## 测试 - 容器 CANN 9.0.0 / Ascend 910B3：Onnx*PluginTest 共 19 用例全 [ PASSED ]。 - CI 增量 --ops=correlation,one_hot,strided_slice --ophost：46 用例（含 19 个 onnx framework 用例）全过。 - pre-commit（clang-format / end-of-file / OAT）全绿。 ## 文档更新不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5325	19 天前
op_api	fix: 责任田算子日志质量整改 Co-authored-by: yifangao<gaoyifan20@huawei.com> # message auto-generated for no-merge-commit merge: !5210 merge fix-log-evaluation-20260829 into master fix: 责任田算子日志质量整改 Created-by: yifangao Commit-by: yifangao Merged-by: cann-robot Description: ## 描述根据日志质量测评结果，对责任田内 HistogramV2、LinSpace、OneHot 和 Range 算子的日志文案进行整改： - HistogramV2：修正 bins 非法值日志语法。 - LinSpace：修正 tilingKey 失败描述，补充 steps 实际值，修正 dtype 及 AICPU 参数校验日志。 - OneHot：修正 numClasses、axis 取值范围的英文表达，完善 coreNum 和 UB size 非法值日志。 - Range：展开 OFF/OFT 缩写，修正 AICore dtype、CalculateOutputSize、CalcRangeTilingParam 和输出维度相关日志。本次仅调整日志文案及格式，不改变算子计算逻辑。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/2985 ## 测试 - 针对变更文件运行 pre-commit，trailing-whitespace、end-of-file-fixer、clang-format、codespell 和 OAT 检查均通过。 - `git diff --check` 检查通过。 - 本次不涉及算子功能逻辑变更，未额外执行算子精度测试。 ## 文档更新无。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：日志质量整改 See merge request: cann/ops-math!5210	26 天前
op_graph	licence update Co-authored-by: daiwei18<daiwei18@huawei.com> # message auto-generated for no-merge-commit merge: !851 merge master into master licence update Created-by: daiwei18 Commit-by: daiwei18 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!851	8 个月前
op_host	fix: 责任田算子日志质量整改 Co-authored-by: yifangao<gaoyifan20@huawei.com> # message auto-generated for no-merge-commit merge: !5210 merge fix-log-evaluation-20260829 into master fix: 责任田算子日志质量整改 Created-by: yifangao Commit-by: yifangao Merged-by: cann-robot Description: ## 描述根据日志质量测评结果，对责任田内 HistogramV2、LinSpace、OneHot 和 Range 算子的日志文案进行整改： - HistogramV2：修正 bins 非法值日志语法。 - LinSpace：修正 tilingKey 失败描述，补充 steps 实际值，修正 dtype 及 AICPU 参数校验日志。 - OneHot：修正 numClasses、axis 取值范围的英文表达，完善 coreNum 和 UB size 非法值日志。 - Range：展开 OFF/OFT 缩写，修正 AICore dtype、CalculateOutputSize、CalcRangeTilingParam 和输出维度相关日志。本次仅调整日志文案及格式，不改变算子计算逻辑。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/2985 ## 测试 - 针对变更文件运行 pre-commit，trailing-whitespace、end-of-file-fixer、clang-format、codespell 和 OAT 检查均通过。 - `git diff --check` 检查通过。 - 本次不涉及算子功能逻辑变更，未额外执行算子精度测试。 ## 文档更新无。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：日志质量整改 See merge request: cann/ops-math!5210	26 天前
op_kernel	[kernel] dynamic_stitch histogram_v2 and one_hot modify simt api Co-authored-by: majiayuan<majiayuan1@huawei.com> # message auto-generated for no-merge-commit merge: !2483 merge master into master [kernel] dynamic_stitch histogram_v2 and one_hot modify simt api Created-by: majiayuan Commit-by: majiayuan Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 根据映射关系，统一将C++风格API替换为C风格API。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [Requirement\|需求建议]: dynamic_stitch, histogram_v2 以及 one_hot SIMT算子整改原有cpp simt接口 [#1492](https://gitcode.com/cann/ops-math/issues/1492) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> SIMT API接口修改前后，反汇编得到的md5码相同。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!2483	4 个月前
tests	refactor: npu_indexing/correlation/one_hot_d onnx 插件迁移至 ParseParamsByOperatorFn 接口 Co-authored-by: clinglai0517<laichangling@huawei.com> # message auto-generated for no-merge-commit merge: !5325 merge fix-onnx-partitionedcall into master refactor: npu_indexing/correlation/one_hot_d onnx 插件迁移至 ParseParamsByOperatorFn 接口 Created-by: clinglai0517 Commit-by: clinglai0517 Merged-by: cann-robot Description: ## 描述将 npu_indexing、correlation、one_hot_d 三个 ONNX 插件从旧 ParseParamsFn(const Message, ge::Operator&) 接口迁移至 ParseParamsByOperatorFn，去除onnx_common.h / ge::onnx:: / NodeProto 依赖（强依赖 ge_onnx.pb.h），插件与 framework UT 编译均脱离 protobuf。 ### 迁移内容 - conversion/strided_slice/framework/npu_indexing_onnx_plugin.cpp + 7 UT ：begins/ends/strides 与五个 mask 经 json 解析。 - common/src/framework/correlation_onnx_plugin.cpp + 5 UT ：groups 属性经 json 解析，保留默认值。 - math/one_hot/framework/one_hot_d_onnx_plugin.cpp + 7 UT ：depth（必需）/ num_classes / on_value / off_value 经 json 解析。 ### 公共工具下沉 onnx_common.h 中所有无 protobuf 依赖的 helper 全部下沉至common/inc/framework/math_onnx_plugin_util.h（protobuf-free）：Vec2Tensor、CreateScalar、DataTypeOnnx、onnx2om_dtype_map、GetOmDtypeFromOnnxDtype、ChangeFormatFromOnnx。onnx_common.h 现仅保留ge_onnx.pb.h 等 include 链供未迁移插件使用，经包含 util 仍提供这些 helper，无破坏。 ### CI 修复 1. scripts/ci/gen_ci_cmd.py：get_op_ut_types 识别 <op>/framework/ 与 <op>/tests/ut/framework/ 变更为 ophost，修复仅改 framework/onnx 插件的 PR 生成 0 条命令、UT 未运行的问题。 2. cmake/ut.cmake：framework 分支 --ops 过滤区分 common 集中目录（按文件名）与独立算子目录（按目录算子名），修复 test__onnx_plugin.cpp 因文件名算子名与目录算子名不一致被 continue 跳过、framework_ut_cases 仅剩 empty.cpp 占位的问题。 ## 关联的Issue Issue #2892 ## 测试 - 容器 CANN 9.0.0 / Ascend 910B3：Onnx*PluginTest 共 19 用例全 [ PASSED ]。 - CI 增量 --ops=correlation,one_hot,strided_slice --ophost：46 用例（含 19 个 onnx framework 用例）全过。 - pre-commit（clang-format / end-of-file / OAT）全绿。 ## 文档更新不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5325	19 天前
CMakeLists.txt	适配PG short_soc_version Co-authored-by: xufeng12121<1074805447@qq.com> # message auto-generated for no-merge-commit merge: !2804 merge mc62 into master 适配PG short_soc_version Created-by: xufeng12121 Commit-by: xufeng12121 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 算子适配PG特性，将math仓 mc62cm12a/MC62CM12A 改成 mc62/MC62 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/1732 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!2804	3 个月前
README.md	md大模型检测低错修改 Co-authored-by: gitcode-chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !5183 merge master into master md大模型检测低错修改 Created-by: gitcode-chenjiao Commit-by: gitcode-chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> md大模型检测低错修改 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [# 2466](https://gitcode.com/cann/ops-math/issues/2446) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> *.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5183	29 天前

OneHot

产品支持情况

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

功能说明

算子功能：对长度为n的输入self，经过one_hot的计算后得到一个元素数量为n*k的输出out，其中k的值为numClasses。
计算公式： $out[i][j]={onValue,self[i]=joffValue,self[i]≠jout[i][j]=\left\{ \begin{aligned} onValue,\quad self[i] = j \\ offValue, \quad self[i] \neq j \end{aligned} \right.$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	表示索引张量。	UINT8、INT32、INT64	ND
depth	输入	类别数。	INT32、INT64	ND
on_value	输入	索引位置的填充值。	FLOAT16、FLOAT、INT8、UINT8、INT32、INT64	ND
off_value	输入	非索引位置的填充值。	FLOAT16、FLOAT、INT8、UINT8、INT32、INT64	ND
axis	输入属性	编码向量的插入维度。	INT32	ND
y	输出	表示one-hot张量。	FLOAT16、FLOAT、INT8、UINT8、INT32、INT64	ND

约束说明

无

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_one_hot	通过aclnnOneHot接口方式调用OneHot算子。