0
代码
代码
Issues
Pull Requests
讨论
Wiki
分析
0
cann-robotcann-robotrefactor: 迁移 experimental 算子 tiling 模块至新公共头文件及命名空间
01df82e2创建于 6 天前历史提交
文件最后提交记录最后更新时间
examples
新增isin_part_v1算子 Co-authored-by: Zoey679<zhangjie679@huawei.com> Co-authored-by: loveRabbit<menghang9@h-partners.com> # message auto-generated for no-merge-commit merge: !989 merge master into master 新增isin_part_v1算子 Created-by: JieZhang679 Commit-by: JieZhang679;loveRabbit;Zoey679 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!9892 个月前
op_host
refactor: 迁移 experimental 算子 tiling 模块至新公共头文件及命名空间 Co-authored-by: songkai111<songkai16@huawei.com> # message auto-generated for no-merge-commit merge: !2974 merge master into master refactor: 迁移 experimental 算子 tiling 模块至新公共头文件及命名空间 Created-by: songkai111 Commit-by: songkai111 Merged-by: cann-robot Description: ## 描述 将 experimental 目录下 22 个算子的 tiling 代码从旧公共头文件及命名空间迁移至新公共头文件及命名空间，统一 tiling 公共接口规范。 ### 改动原因 experimental 算子的 tiling 代码依赖了旧的公共头文件（tiling_base_util.h、math_tiling_templates_registry.h、tiling_base_class.h）及 Ops::Base 命名空间，需要迁移至新的公共头文件（tiling_util.h、tiling_templates_registry.h、tiling_base.h）及 Ops::Math::OpTiling 命名空间，以统一 tiling 公共接口规范。 ### 改动方法 1. 替换 #include 指令： - "op_host/tiling_base_util.h" → "op_host/tiling_util.h"（20 个文件） - "op_host/math_tiling_templates_registry.h" → "op_host/tiling_templates_registry.h"（2 个文件） - "op_host/tiling_base_class.h" → "op_host/tiling_base.h"（1 个文件） 2. 命名空间及 API 迁移： - Ops::Base::EnsureNotScalar() → Ops::Math::OpTiling::EnsureNotScalar() - Ops::Base::IsRegbaseSocVersion() → Ops::Math::OpTiling::IsRegbaseSocVersion() - using namespace Ops::Base → using namespace Ops::Math::OpTiling（2 个文件） ### 涉及算子（22 个文件） - **conversion 算子**：col2_im、im2_col - **math 算子**：add_v2、atan、ceil_v2、cross、dot_v2、floor_div、floor_mod、ger_v2、greater_equal（arch35）、isin_part_v1、minimum、ones_like、real、select_v3、sinh、softsign（arch35）、square_v2、trilu（arch35）、trunc_v2 ## 关联的Issue - #1679 - https://gitcode.com/cann/ops-math/issues/1679 ## 测试 通过编译验证及算子冒烟测试。 ## 文档更新 无文档更新。 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：重构 - 迁移 tiling 公共头文件及命名空间 See merge request: cann/ops-math!29746 天前
op_kernel
新增isin_part_v1算子 Co-authored-by: Zoey679<zhangjie679@huawei.com> Co-authored-by: loveRabbit<menghang9@h-partners.com> # message auto-generated for no-merge-commit merge: !989 merge master into master 新增isin_part_v1算子 Created-by: JieZhang679 Commit-by: JieZhang679;loveRabbit;Zoey679 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!9892 个月前
CMakeLists.txt
新增isin_part_v1算子 Co-authored-by: Zoey679<zhangjie679@huawei.com> Co-authored-by: loveRabbit<menghang9@h-partners.com> # message auto-generated for no-merge-commit merge: !989 merge master into master 新增isin_part_v1算子 Created-by: JieZhang679 Commit-by: JieZhang679;loveRabbit;Zoey679 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!9892 个月前
README.md
doc tools 工具扫描问题修改 Co-authored-by: gitee-yanglulu<yanglulul@h-partners.com> # message auto-generated for no-merge-commit merge: !1938 merge master into master doc tools 工具扫描问题修改 Created-by: gitee-yanglulu Commit-by: gitee-yanglulu Merged-by: cann-robot Description: doc tools 工具扫描问题修改 See merge request: cann/ops-math!19382 个月前
README.md

IsinPart

产品支持情况

产品 是否支持
Atlas A2 训练系列产品/Atlas 800I A2 推理产品 √

功能说明

  • 算子功能：isin功能核心part，结合cat、sort实现isin功能

  • isin算子功能：判断给定张量elements的每个元素是否在另一个张量test_elements中出现，并返回一个布尔类型张量，形状与给定张量一致。

  • isin算子完整实现：

elementsNum=elements.length()combine_elements=cat(elements,test_elements)value,index=sort(combine_elements)z=isin_part_v1(value,index,elementsNum)elementsNum = elements.length() \\ combine\_elements = cat(elements, test\_elements) \\ value, index = sort(combine\_elements) \\ z = isin\_part\_v1(value, index, elementsNum) \\

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
value 输入 elements与test_elements拼接排序后的张量值 FLOAT、INT32 ND
index 输入 elements与test_elements拼接排序后的张量索引 INT32 ND
elementsNum 输入 elements元素数量(标量) INT64、INT32 ND
z 输出 isin操作后的输出张量，形状由elementsNum参数决定 BOOL ND

约束说明

无

调用说明

调用方式 调用样例 说明
aclnn调用 test_aclnn_isin_part_v1.cpp 通过[test_aclnn_isin_part_v1]接口方式调用IsinPartV1算子。