已关闭
[Requirement|需求建议]: trans_data算子新增ascend350平台支持 #3166
马琦钧创建于  9月14日关闭于  9月15日
马琦钧
马琦钧成员
9月14日 创建

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 实现:

  1. conversion/trans_data/CMakeLists.txt:SUPPORT_COMPUTE_UNIT / SUPPORT_TILING_DIR 增加 ascend350(tiling 复用 arch35 目录)
  2. conversion/trans_data/op_host/trans_data_def.cpp:AICore 增加 ascend350 配置,复用 ascend950 的 aicore_config
  3. 新增 op_host/config/ascend350/trans_data_binary.json、trans_data_simplified_key.ini(与 ascend950 保持一致)
  4. 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 平台一致

likedislike
陈思
陈思成员
9月14日 评论:

/assign @maqijun

likedislike
CANN-robotCANN-robot成员
9月14日 将 maqijun 设为负责人
马琦钧
马琦钧成员
9月15日 评论:

https://gitcode.com/cann/ops-math/pull/5560/ 这个 pr 已上库trans_data适配ascend350,本issue关闭

likedislike
马琦钧
马琦钧成员
9月15日 评论:

/close

likedislike
CANN-robotCANN-robot成员
9月15日 issue状态由 进行中 改变为 已完成
CANN-robotCANN-robot成员
9月15日 关闭了 issue