已关闭
[Requirement|需求建议]: trans_data算子新增ascend350平台支持 #3166
马琦钧创建于 20 天前关闭于 19 天前
20 天前 将 maqijun 设为负责人
马琦钧
19 天前 评论:
19 天前 评论:
https://gitcode.com/cann/ops-math/pull/5560/ 这个 pr 已上库trans_data适配ascend350,本issue关闭


马琦钧
19 天前 评论:
19 天前 评论:
/close


19 天前 issue状态由 进行中 改变为 已完成
19 天前 关闭了 issue
Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
一、背景信息 (必填)
trans_data 算子新增 ascend350 平台支持,参考 cann/ops-nn!5116(matmul support ascend350)的适配方式。
二、价值/作用 (必填)
使 TransData 算子在 ascend350 平台可用,支撑 ascend350 平台上的格式转换场景(ND/NZ 等格式互转)。
三、设计方案 (必填)
复用 ascend950 的 arch35 实现:
conversion/trans_data/CMakeLists.txt:SUPPORT_COMPUTE_UNIT/SUPPORT_TILING_DIR增加 ascend350(tiling 复用 arch35 目录)conversion/trans_data/op_host/trans_data_def.cpp:AICore 增加 ascend350 配置,复用 ascend950 的 aicore_configop_host/config/ascend350/trans_data_binary.json、trans_data_simplified_key.ini(与 ascend950 保持一致)scripts/kernel/binary_config/ascendc_config.json:TransData 增加 ascend350 编译单元及编译选项(与 ascend950 保持一致)3.1 使能方式(涉及哪些框架:如Aclnn直调、Pytorch训练等)
Aclnn 直调 / GE 图模式
3.2 总体设计
3.2.1 算子支持的数据类型
与 ascend950 平台已支持的数据类型一致
3.2.2 host侧设计
复用 arch35 tiling(transdata_tiling_arch35)
3.2.3 kernel侧设计
复用 trans_data_apt(AscendC)kernel,ascend350 与 ascend950 使用相同的编译选项
3.3 支持硬件
ascend350(Ascend350_355e)
3.4 算子约束限制
与 ascend950 平台一致