已关闭
[Requirement|需求建议]: trans_data算子新增ascend350平台支持 #3166
马琦钧创建于  20 天前关闭于  19 天前
马琦钧
马琦钧成员
20 天前 创建

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
陈思
陈思成员
20 天前 评论:

/assign @maqijun

likedislike
CANN-robotCANN-robot成员
20 天前 将 maqijun 设为负责人
马琦钧
马琦钧成员
19 天前 评论:

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

likedislike
马琦钧
马琦钧成员
19 天前 评论:

/close

likedislike
CANN-robotCANN-robot成员
19 天前 issue状态由 进行中 改变为 已完成
CANN-robotCANN-robot成员
19 天前 关闭了 issue