| Files | Last commit | Last update |
|---|---|---|
feat(ones_like): support uint32 and uint64 via dtype views Co-authored-by: zhouyuanhang30083383<zhouyuanhang4@h-partners.com> # message auto-generated for no-merge-commit merge: !7014 merge feat/aclnn-inplace-one-dtype-view into master feat(ones_like): support uint32 and uint64 via dtype views Created-by: m0_55003149 Commit-by: zhouyuanhang30083383 Merged-by: cann-robot Description: ## 描述 aclnnInplaceOne 支持 uint32、uint64,在 ACLNN 接口层将 uint32、uint64 映射为同宽的 int32、int64 元数据视图。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/4515 ## 测试 - 精度:256 条 ACLNN 用例全部通过 - 性能:uint32/int32 中位耗时比为 0.9785-1.0121,uint64/int64 为 0.9990-1.0013; - 内存:uint32/int32 workspace 均为 16777728 B;uint64/int64 均为 0 B;同宽 I/O 字节逐例一致。 - op_api UT:l2_ones_test.* 共 34 条全部通过,其中新增 9 条 ## 文档更新 - math/ones_like/docs/aclnnInplaceOne.md - math/ones_like/docs/aclnnInplaceOne_en.md ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!7014 | 9 days ago | |
abs等算子A5实现 Co-authored-by: ly_cann_coder<liyao152@huawei.com> # message auto-generated for no-merge-commit merge: !495 merge develop into master abs等算子A5实现 Created-by: ly_cann_coder Commit-by: ly_cann_coder Merged-by: cann-robot Description: ## 描述 新增以下算子的A5实现 |序号| 算子 | 功能 | |--|--|--| |1| abs | 为输入张量的每一个元素取绝对值 | |2| add | 对输入完成相加操作 | |3| addcdiv | 张量运算函数,用于执行乘除加组合操作,将张量除法(带缩放)+ 张量加法合并为单个操作 | |4| addcmul | 张量运算函数,用于执行乘除加组合操作,将张量乘法(带缩放)+ 张量加法合并为单个操作 | |5| axpy | 源操作数中每个元素与标量求积后和目的操作数中的对应元素相加 | |6| bincount | 计算非负整数数组中每个数的频率 | |7| ceil | 返回输入tensor中每个元素向上取整的结果 | |8| clip_by_value_v2 | 将输入的所有元素限制在一定范围内 | |9| concat | 用于沿指定维度将多个输入 Tensor 进行拼接,输出包含所有输入数据按顺序拼接后的 Tensor | |10| concat_v2 | 用于沿指定维度将多个输入 Tensor 进行拼接,输出包含所有输入数据按顺序拼接后的 Tensor | |11| equal | 计算两个Tensor是否有相同的大小和元素,返回一个Bool类型 | |12|exp|返回一个新的张量,该张量的每个元素都是输入张量对应元素的指数| |13| fill | 对张量进行填充操作,支持非连续的Tensor操作 | |14| floor | 返回输入Tensor中每个元素向下取整,并将结果回填到输入Tensor中 | |15| floor_div | 完成除法计算,对余数向下取整 | |16| floor_mod | 用于计算两个数的模 | |17| is_neg_inf | 判断输入张量的元素是否为负无穷 | |18| is_pos_inf | 判断输入张量的元素是否为正无穷 | |19| less | 判断输入self中的每个元素是否小于输入other的值,返回一个Bool类型的Tensor | |20| less_equal | 判断输入self中的元素值是否小于等于other的值,并将self的每个元素的值与other值的比较结果写入out中 | |21| logical_and | 对两个输入张量的对应元素执行「与逻辑」判断,输出布尔型张量(True/False) | |22| log | 对输入张量x的元素,逐元素进行对数计算,并将结果保存到输出张量y中 | |23| not_equal | 逐元素比较两个输入张量是否不相等 | |24| ones_like | 返回形状和类型相同的张量,所有元素都设置为1 | |25|sign|对输入的tensor逐元素进行Sign符号函数的运算并输出结果tensor| |26| sqrt | 完成非负数平方根计算,负数情况返回nan | |27| sqrt_grad | 完成梯度平方根计算 | |28| square | 为输入张量的每一个元素计算平方值 | |29| squared_difference | 为第一个输入张量减去第二个输入张量,并计算其平方值 | |30| tanh | 激活函数,返回与输入tensor shape相同的tensor,对输入tensor进行elementwise的计算 | |31| tril | 将输入的self张量的最后二维(按shape从左向右数)沿对角线的右上部分置零 | |32| triu | 将输入的self张量的最后二维(按shape从左向右数)沿对角线的左下部分置零 | |33| zeros_like | 将张量填充为全零 | ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!495 | 10 months ago | |
迁移开源TF插件--math Co-authored-by: yaochen<yaochen15@huawei.com> # message auto-generated for no-merge-commit merge: !4308 merge transfer-math into master 迁移开源TF插件--math Created-by: nextyale Commit-by: yaochen Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他,请描述:迁移TF插件 See merge request: cann/ops-math!4308 | 3 months ago | |
ones_like新增int64dtype支持 Co-authored-by: wangrui<wangrui124@huawei.com> # message auto-generated for no-merge-commit merge: !7158 merge feat/ones-like-int64 into master ones_like新增int64dtype支持 Created-by: wangrui_ Commit-by: wangrui Merged-by: cann-robot Description: ## 描述 为 OnesLike 算子新增 **INT64** dtype 支持,使 int64 输入在 ascend950/ascend350/ascend960dt 上 走 AICore(AIV 向量核)执行路径。此前 int64 仅支持 AICPU 回退路径,性能受限。 **改动方法**(12 个文件,+118/-17 行,kernel 零改动): - op_host/ones_like_def.cpp:Input/Output 的 DataType/Format/UnknownShapeFormat 列表追加 DT_INT64 - op_kernel/arch35/ones_like_tiling_key.h:新增 ONES_LIKE_TPL_INT64 tiling key 及对应 DECL/SEL 声明 - op_host/arch35/ones_like_tiling_arch35.cpp:SetTilingData/CalcInputDtype/RunTiling 增加 int64 分支(DoTiling<OnesLikeDAG<int64_t>>) - op_api/ones_like.cpp:AICore dtype 支持列表加入 DT_INT64 - 三个芯片目录 ones_like_binary.json 各新增 OnesLike_INT64 二进制条目 - UT/st/README 同步更新 kernel 侧无需修改:DAG 模板自动实例化 OnesLikeDAG<int64_t>,实现方式与 conversion/zeros_like 既有 int64 支持一致。老 910 芯片保持 AICPU 回退,行为不变。与 master 上 uint32/uint64 dtype views 特性互补。 ## 关联的Issue Closes #<[issue编号](https://gitcode.com/cann/ops-math/issues/4546)> ## 测试 - **编译**:bash build.sh --pkg --soc=ascend950 --ops=ones_like,正常产出 OnesLike_INT64_high_performance.o(AIV/VectorCore) - **UT**:op_host UT 8/8 PASSED(test_tiling_int64_001 翻转为 SUCCESS,expectTilingKey=17); op_api UT 25/25 PASSED,无回归 - **上板验证(Ascend 950PR)**:int64 shape=(8,16) 与 (4096,4096),输出全量元素 = 1,结果正确; debug 日志确认走 OnesLikeAiCore + DT_INT64(AICore 新路径,非 AICPU 回退) - 其余 7 种 dtype 回归通过,无影响 ## 文档更新 - math/ones_like/README.md:参数说明 dtype 表(x/y)追加 INT64 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ √] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!7158 | 8 days ago | |
abs等算子A5实现 Co-authored-by: ly_cann_coder<liyao152@huawei.com> # message auto-generated for no-merge-commit merge: !495 merge develop into master abs等算子A5实现 Created-by: ly_cann_coder Commit-by: ly_cann_coder Merged-by: cann-robot Description: ## 描述 新增以下算子的A5实现 |序号| 算子 | 功能 | |--|--|--| |1| abs | 为输入张量的每一个元素取绝对值 | |2| add | 对输入完成相加操作 | |3| addcdiv | 张量运算函数,用于执行乘除加组合操作,将张量除法(带缩放)+ 张量加法合并为单个操作 | |4| addcmul | 张量运算函数,用于执行乘除加组合操作,将张量乘法(带缩放)+ 张量加法合并为单个操作 | |5| axpy | 源操作数中每个元素与标量求积后和目的操作数中的对应元素相加 | |6| bincount | 计算非负整数数组中每个数的频率 | |7| ceil | 返回输入tensor中每个元素向上取整的结果 | |8| clip_by_value_v2 | 将输入的所有元素限制在一定范围内 | |9| concat | 用于沿指定维度将多个输入 Tensor 进行拼接,输出包含所有输入数据按顺序拼接后的 Tensor | |10| concat_v2 | 用于沿指定维度将多个输入 Tensor 进行拼接,输出包含所有输入数据按顺序拼接后的 Tensor | |11| equal | 计算两个Tensor是否有相同的大小和元素,返回一个Bool类型 | |12|exp|返回一个新的张量,该张量的每个元素都是输入张量对应元素的指数| |13| fill | 对张量进行填充操作,支持非连续的Tensor操作 | |14| floor | 返回输入Tensor中每个元素向下取整,并将结果回填到输入Tensor中 | |15| floor_div | 完成除法计算,对余数向下取整 | |16| floor_mod | 用于计算两个数的模 | |17| is_neg_inf | 判断输入张量的元素是否为负无穷 | |18| is_pos_inf | 判断输入张量的元素是否为正无穷 | |19| less | 判断输入self中的每个元素是否小于输入other的值,返回一个Bool类型的Tensor | |20| less_equal | 判断输入self中的元素值是否小于等于other的值,并将self的每个元素的值与other值的比较结果写入out中 | |21| logical_and | 对两个输入张量的对应元素执行「与逻辑」判断,输出布尔型张量(True/False) | |22| log | 对输入张量x的元素,逐元素进行对数计算,并将结果保存到输出张量y中 | |23| not_equal | 逐元素比较两个输入张量是否不相等 | |24| ones_like | 返回形状和类型相同的张量,所有元素都设置为1 | |25|sign|对输入的tensor逐元素进行Sign符号函数的运算并输出结果tensor| |26| sqrt | 完成非负数平方根计算,负数情况返回nan | |27| sqrt_grad | 完成梯度平方根计算 | |28| square | 为输入张量的每一个元素计算平方值 | |29| squared_difference | 为第一个输入张量减去第二个输入张量,并计算其平方值 | |30| tanh | 激活函数,返回与输入tensor shape相同的tensor,对输入tensor进行elementwise的计算 | |31| tril | 将输入的self张量的最后二维(按shape从左向右数)沿对角线的右上部分置零 | |32| triu | 将输入的self张量的最后二维(按shape从左向右数)沿对角线的左下部分置零 | |33| zeros_like | 将张量填充为全零 | ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!495 | 10 months ago | |
ones_like新增int64dtype支持 Co-authored-by: wangrui<wangrui124@huawei.com> # message auto-generated for no-merge-commit merge: !7158 merge feat/ones-like-int64 into master ones_like新增int64dtype支持 Created-by: wangrui_ Commit-by: wangrui Merged-by: cann-robot Description: ## 描述 为 OnesLike 算子新增 **INT64** dtype 支持,使 int64 输入在 ascend950/ascend350/ascend960dt 上 走 AICore(AIV 向量核)执行路径。此前 int64 仅支持 AICPU 回退路径,性能受限。 **改动方法**(12 个文件,+118/-17 行,kernel 零改动): - op_host/ones_like_def.cpp:Input/Output 的 DataType/Format/UnknownShapeFormat 列表追加 DT_INT64 - op_kernel/arch35/ones_like_tiling_key.h:新增 ONES_LIKE_TPL_INT64 tiling key 及对应 DECL/SEL 声明 - op_host/arch35/ones_like_tiling_arch35.cpp:SetTilingData/CalcInputDtype/RunTiling 增加 int64 分支(DoTiling<OnesLikeDAG<int64_t>>) - op_api/ones_like.cpp:AICore dtype 支持列表加入 DT_INT64 - 三个芯片目录 ones_like_binary.json 各新增 OnesLike_INT64 二进制条目 - UT/st/README 同步更新 kernel 侧无需修改:DAG 模板自动实例化 OnesLikeDAG<int64_t>,实现方式与 conversion/zeros_like 既有 int64 支持一致。老 910 芯片保持 AICPU 回退,行为不变。与 master 上 uint32/uint64 dtype views 特性互补。 ## 关联的Issue Closes #<[issue编号](https://gitcode.com/cann/ops-math/issues/4546)> ## 测试 - **编译**:bash build.sh --pkg --soc=ascend950 --ops=ones_like,正常产出 OnesLike_INT64_high_performance.o(AIV/VectorCore) - **UT**:op_host UT 8/8 PASSED(test_tiling_int64_001 翻转为 SUCCESS,expectTilingKey=17); op_api UT 25/25 PASSED,无回归 - **上板验证(Ascend 950PR)**:int64 shape=(8,16) 与 (4096,4096),输出全量元素 = 1,结果正确; debug 日志确认走 OnesLikeAiCore + DT_INT64(AICore 新路径,非 AICPU 回退) - 其余 7 种 dtype 回归通过,无影响 ## 文档更新 - math/ones_like/README.md:参数说明 dtype 表(x/y)追加 INT64 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ √] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!7158 | 8 days ago | |
ones_like新增int64dtype支持 Co-authored-by: wangrui<wangrui124@huawei.com> # message auto-generated for no-merge-commit merge: !7158 merge feat/ones-like-int64 into master ones_like新增int64dtype支持 Created-by: wangrui_ Commit-by: wangrui Merged-by: cann-robot Description: ## 描述 为 OnesLike 算子新增 **INT64** dtype 支持,使 int64 输入在 ascend950/ascend350/ascend960dt 上 走 AICore(AIV 向量核)执行路径。此前 int64 仅支持 AICPU 回退路径,性能受限。 **改动方法**(12 个文件,+118/-17 行,kernel 零改动): - op_host/ones_like_def.cpp:Input/Output 的 DataType/Format/UnknownShapeFormat 列表追加 DT_INT64 - op_kernel/arch35/ones_like_tiling_key.h:新增 ONES_LIKE_TPL_INT64 tiling key 及对应 DECL/SEL 声明 - op_host/arch35/ones_like_tiling_arch35.cpp:SetTilingData/CalcInputDtype/RunTiling 增加 int64 分支(DoTiling<OnesLikeDAG<int64_t>>) - op_api/ones_like.cpp:AICore dtype 支持列表加入 DT_INT64 - 三个芯片目录 ones_like_binary.json 各新增 OnesLike_INT64 二进制条目 - UT/st/README 同步更新 kernel 侧无需修改:DAG 模板自动实例化 OnesLikeDAG<int64_t>,实现方式与 conversion/zeros_like 既有 int64 支持一致。老 910 芯片保持 AICPU 回退,行为不变。与 master 上 uint32/uint64 dtype views 特性互补。 ## 关联的Issue Closes #<[issue编号](https://gitcode.com/cann/ops-math/issues/4546)> ## 测试 - **编译**:bash build.sh --pkg --soc=ascend950 --ops=ones_like,正常产出 OnesLike_INT64_high_performance.o(AIV/VectorCore) - **UT**:op_host UT 8/8 PASSED(test_tiling_int64_001 翻转为 SUCCESS,expectTilingKey=17); op_api UT 25/25 PASSED,无回归 - **上板验证(Ascend 950PR)**:int64 shape=(8,16) 与 (4096,4096),输出全量元素 = 1,结果正确; debug 日志确认走 OnesLikeAiCore + DT_INT64(AICore 新路径,非 AICPU 回退) - 其余 7 种 dtype 回归通过,无影响 ## 文档更新 - math/ones_like/README.md:参数说明 dtype 表(x/y)追加 INT64 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ √] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!7158 | 8 days ago | |
add aicpu:coordinates_1d_to_2d, ones_like Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !2785 merge dev_extra into master add aicpu:coordinates_1d_to_2d, ones_like Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述 新增三个 AICPU 算子实现: - **coordinates_1d_to_2d**: 将1D坐标转换为2D坐标,根据shape信息计算行列索引 - **ones_like**: 返回形状和类型相同的张量,所有元素都设置为1 ## 关联的Issue #1584 ## 测试 根据代码变更,测试场景如下: 1. **编译构建测试** - 执行 cmake 配置和编译,验证新增算子能正确构建 - 验证 CMakeLists.txt 配置正确 2. **算子加载测试** - 验证算子接口注册成功 - 测试 GE IR 模式加载 3. **单元测试** - 运行 tests/ut/op_kernel_aicpu 目录下的单元测试 - 验证算子计算逻辑正确性 4. **功能验证** - 验证 coordinates_1d_to_2d 坐标转换功能 - 验证 ones_like 张量生成功能 ## 文档更新 - 新增 conversion/coordinates_1d_to_2d/README.md - 新增算子功能说明和数据类型支持列表 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!2785 | 5 months ago | |
ones_like新增int64dtype支持 Co-authored-by: wangrui<wangrui124@huawei.com> # message auto-generated for no-merge-commit merge: !7158 merge feat/ones-like-int64 into master ones_like新增int64dtype支持 Created-by: wangrui_ Commit-by: wangrui Merged-by: cann-robot Description: ## 描述 为 OnesLike 算子新增 **INT64** dtype 支持,使 int64 输入在 ascend950/ascend350/ascend960dt 上 走 AICore(AIV 向量核)执行路径。此前 int64 仅支持 AICPU 回退路径,性能受限。 **改动方法**(12 个文件,+118/-17 行,kernel 零改动): - op_host/ones_like_def.cpp:Input/Output 的 DataType/Format/UnknownShapeFormat 列表追加 DT_INT64 - op_kernel/arch35/ones_like_tiling_key.h:新增 ONES_LIKE_TPL_INT64 tiling key 及对应 DECL/SEL 声明 - op_host/arch35/ones_like_tiling_arch35.cpp:SetTilingData/CalcInputDtype/RunTiling 增加 int64 分支(DoTiling<OnesLikeDAG<int64_t>>) - op_api/ones_like.cpp:AICore dtype 支持列表加入 DT_INT64 - 三个芯片目录 ones_like_binary.json 各新增 OnesLike_INT64 二进制条目 - UT/st/README 同步更新 kernel 侧无需修改:DAG 模板自动实例化 OnesLikeDAG<int64_t>,实现方式与 conversion/zeros_like 既有 int64 支持一致。老 910 芯片保持 AICPU 回退,行为不变。与 master 上 uint32/uint64 dtype views 特性互补。 ## 关联的Issue Closes #<[issue编号](https://gitcode.com/cann/ops-math/issues/4546)> ## 测试 - **编译**:bash build.sh --pkg --soc=ascend950 --ops=ones_like,正常产出 OnesLike_INT64_high_performance.o(AIV/VectorCore) - **UT**:op_host UT 8/8 PASSED(test_tiling_int64_001 翻转为 SUCCESS,expectTilingKey=17); op_api UT 25/25 PASSED,无回归 - **上板验证(Ascend 950PR)**:int64 shape=(8,16) 与 (4096,4096),输出全量元素 = 1,结果正确; debug 日志确认走 OnesLikeAiCore + DT_INT64(AICore 新路径,非 AICPU 回退) - 其余 7 种 dtype 回归通过,无影响 ## 文档更新 - math/ones_like/README.md:参数说明 dtype 表(x/y)追加 INT64 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ √] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!7158 | 8 days ago | |
feat: masked_select_v3/zeros_like/ones_like/zero_op 支持 Ascend960 Co-authored-by: zhu-xun00613388<zhuxun7@huawei.com> # message auto-generated for no-merge-commit merge: !6264 merge v121_master into master feat: masked_select_v3/zeros_like/ones_like/zero_op 支持 Ascend960 Created-by: zhu-xun Commit-by: zhu-xun00613388 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> feat: masked_select_v3/zeros_like/ones_like/zero_op 支持 Ascend960 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!6264 | 28 days ago | |
ones_like新增int64dtype支持 Co-authored-by: wangrui<wangrui124@huawei.com> # message auto-generated for no-merge-commit merge: !7158 merge feat/ones-like-int64 into master ones_like新增int64dtype支持 Created-by: wangrui_ Commit-by: wangrui Merged-by: cann-robot Description: ## 描述 为 OnesLike 算子新增 **INT64** dtype 支持,使 int64 输入在 ascend950/ascend350/ascend960dt 上 走 AICore(AIV 向量核)执行路径。此前 int64 仅支持 AICPU 回退路径,性能受限。 **改动方法**(12 个文件,+118/-17 行,kernel 零改动): - op_host/ones_like_def.cpp:Input/Output 的 DataType/Format/UnknownShapeFormat 列表追加 DT_INT64 - op_kernel/arch35/ones_like_tiling_key.h:新增 ONES_LIKE_TPL_INT64 tiling key 及对应 DECL/SEL 声明 - op_host/arch35/ones_like_tiling_arch35.cpp:SetTilingData/CalcInputDtype/RunTiling 增加 int64 分支(DoTiling<OnesLikeDAG<int64_t>>) - op_api/ones_like.cpp:AICore dtype 支持列表加入 DT_INT64 - 三个芯片目录 ones_like_binary.json 各新增 OnesLike_INT64 二进制条目 - UT/st/README 同步更新 kernel 侧无需修改:DAG 模板自动实例化 OnesLikeDAG<int64_t>,实现方式与 conversion/zeros_like 既有 int64 支持一致。老 910 芯片保持 AICPU 回退,行为不变。与 master 上 uint32/uint64 dtype views 特性互补。 ## 关联的Issue Closes #<[issue编号](https://gitcode.com/cann/ops-math/issues/4546)> ## 测试 - **编译**:bash build.sh --pkg --soc=ascend950 --ops=ones_like,正常产出 OnesLike_INT64_high_performance.o(AIV/VectorCore) - **UT**:op_host UT 8/8 PASSED(test_tiling_int64_001 翻转为 SUCCESS,expectTilingKey=17); op_api UT 25/25 PASSED,无回归 - **上板验证(Ascend 950PR)**:int64 shape=(8,16) 与 (4096,4096),输出全量元素 = 1,结果正确; debug 日志确认走 OnesLikeAiCore + DT_INT64(AICore 新路径,非 AICPU 回退) - 其余 7 种 dtype 回归通过,无影响 ## 文档更新 - math/ones_like/README.md:参数说明 dtype 表(x/y)追加 INT64 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ √] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!7158 | 8 days ago | |
docs: add English translations for markdown files Co-authored-by: jenny_xiang<jenny.xiang@huawei.com> # message auto-generated for no-merge-commit merge: !7113 merge docs/add-english-translations into master docs: add English translations for markdown files Created-by: jenny_xiang Commit-by: jenny_xiang Merged-by: cann-robot Description: ## 描述 新增仓库Markdown文档英文译稿(第二批,共409个文件):第一批(已合入)之外的其余全部译稿,覆盖conversion、examples、ge、math、random各类算子README译稿,以及CHANGELOG_en.md、.gitcode PR模板英文版等。 本批内全部仓内链接均指向master已有文件(含第一批已合入的21+2篇译稿),合入后本仓432篇英文文档全部就位,链接完整生效。 本批同时包含4篇基于最新源文档的增量更新(math/ada_cast、math/prod_force_se_a、math/sinkhorn、math/xlogy,对应上游近期文档变更)。 ## 关联的Issue 无 ## 测试 文档类变更,无需算子测试;已通过链接有效性校验(本批全部相对链接仅指向master已有文件,无指向本PR外未提交文件的链接),并完成逐篇翻译规范自查与语义校对(含支持状态表、限定词、术语、数值符号核对)。 ## 文档更新 新增409个英文译稿文件(*_en.md / .en-US.md),不修改任何现有文件。 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!7113 | 9 days ago |
OnesLike
English | 简体中文
Product Support Status
| Product | Supported |
|---|---|
| Ascend 950PR&950DT products | √ |
| Atlas A3 products | √ |
| Atlas A2 products | √ |
| Atlas 200I/500 A2 inference products | √ |
| Atlas inference products | √ |
| Atlas training products | √ |
Function Description
-
Operator function: returns a tensor with the same shape and type, with all elements set to 1.
-
Computation formula:
selfRefi=1 {selfRef}_{i} = 1 selfRefi=1
-
Example:
Input input: tensor([[1, 2], [3, 4]]) Output output: tensor([[1, 1], [1, 1]])
Parameter Description
| Parameter Name | Input/Output/Attribute | Description | Data Type | Data Format |
|---|---|---|---|---|
| x | Input | Input parameter for the oneslike computation, which is selfRef in the formula. | FLOAT, FLOAT16, BFLOAT16, INT8, INT32, UINT8, BOOL | ND |
| y | Output | Output parameter for the oneslike computation. The selfRef tensor is filled with all 1s. | FLOAT, FLOAT16, BFLOAT16, INT8, INT32, UINT8, BOOL | ND |
Constraint Description
None
Invocation Description
| Invocation Method | Invocation Sample | Description |
|---|---|---|
| aclnn invocation | test_aclnn_inplace_one | Invokes the OnesLike operator through the aclnnInplaceOne API. |