Star0
0
Code
Code
Issues
Pull Requests
Actions
Discussion
Wiki
Analysis
Star0
0
cann-robotcann-robotKirin X90/9030支持Cast/Mul/Transpose/Add/ConcatD/StridedSlice算子
32c65263Add on 13 days agoHistory
FilesLast commitLast update
arch31
Kirin X90/9030支持Cast/Mul/Transpose/Add/ConcatD/StridedSlice算子 Co-authored-by: chenlong1<chenlong1@huawei.com> Co-authored-by: jinzy1999<jinzy@pku.edu.cn> # message auto-generated for no-merge-commit merge: !6581 merge master into master Kirin X90/9030支持Cast/Mul/Transpose/Add/ConcatD/StridedSlice算子 Created-by: chenlong1 Commit-by: chenlong1;jinzy1999 Merged-by: cann-robot Description: ## 描述 Kirin X90/9030支持Cast/Mul/Transpose/Add/ConcatD/StridedSlice算子 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/3790 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6581
13 days ago
arch35
feat: slice、strided_slice、strided_slice_v2、strided_slice_v3 算子新增ascend960dt平台支持 Co-authored-by: xiapengcheng5<xiapengcheng5@huawei.com> Co-authored-by: yaolu_happy<yaolu15@huawei.com> # message auto-generated for no-merge-commit merge: !6279 merge master into master feat: slice、strided_slice、strided_slice_v2、strided_slice_v3 算子新增ascend960dt平台支持 Created-by: xiapengcheng5 Commit-by: yaolu_happy;xiapengcheng5 Merged-by: cann-robot Description: ## 描述 为 slice、strided_slice、strided_slice_v2、strided_slice_v3 四个算子新增 ascend960dt 平台支持，使其可在 ascend960dt（Ascend960DT_968B7）芯片上完成编译与调用。主要改动： ## 算子定义与编译配置 各算子 op_host 定义文件（slice_def.cpp、strided_slice_def.cpp、strided_slice_v2_def.cpp、strided_slice_v3_def.cpp）通过 AICore().AddConfig("ascend960dt", ...) 注册平台配置 各算子 CMakeLists.txt 的 SUPPORT_COMPUTE_UNIT 增加 "ascend960dt"，tiling 目录沿用 arch35；get_shape 的 kernel 源码 COMPUTE_UNITS 同步增加 ascend960dt 新增各算子 ascend960dt 平台二进制编译配置 op_host/config/ascend960dt/*_binary.json 与 *_simplified_key.ini（simplified_key_mode=0） ascendc_config.json 中 slice、strided_slice、strided_slice_v2、strided_slice_v3 的 compute_units 增加 ascend960dt，并配置对应编译选项 ## op_api 适配 slice.cpp 的 IsSliceV2ARAFullLoadSupport 增加 NpuArch::DAV_9201 分支，数据类型支持列表新增 （FLOAT4_E1M2、FLOAT4_E2M1） strided_slice.cpp 的 isAiCoreSupport 增加 NpuArch::DAV_9201 分支，数据类型支持列表新增 （FLOAT4_E1M2、FLOAT4_E2M1） ## 关联的Issue [#3582](https://gitcode.com/cann/ops-math/issues/3582) <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 已通过 pre-commit 检查（clang-format 格式、codespell 拼写、文件末尾换行整改，见提交记录） ## 文档更新 无 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6279
27 days ago
strided_slice_apt.cpp
Kirin X90/9030支持Cast/Mul/Transpose/Add/ConcatD/StridedSlice算子 Co-authored-by: chenlong1<chenlong1@huawei.com> Co-authored-by: jinzy1999<jinzy@pku.edu.cn> # message auto-generated for no-merge-commit merge: !6581 merge master into master Kirin X90/9030支持Cast/Mul/Transpose/Add/ConcatD/StridedSlice算子 Created-by: chenlong1 Commit-by: chenlong1;jinzy1999 Merged-by: cann-robot Description: ## 描述 Kirin X90/9030支持Cast/Mul/Transpose/Add/ConcatD/StridedSlice算子 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/3790 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6581
13 days ago