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


马琦钧
9月15日 评论:
9月15日 评论:
/close


9月15日 issue状态由 进行中 改变为 已完成
9月15日 关闭了 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 平台一致