已关闭
[Requirement|需求建议]: causal_conv1d 系列算子适配 ascend960平台 #6199
YanWenkai创建于  15 天前关闭于  14 天前
YanWenkai
15 天前 创建

Backgroud(背景信息)

causal_conv1d 系列算子(CausalConv1d、FusedCausalConv1d、InplaceFusedCausalConv1d、MaskedCausalConv1d、MaskedCausalConv1dBackward)此前仅配置了 ascend950 / ascend350 等平台,尚未适配 ascend960平台,需要补齐该平台的算子定义与编译配置。

Benefit / Necessity(价值/作用)

使 causal_conv1d 系列算子可在 ascend960dt 平台上编译与运行,扩展该系列算子的芯片支持范围。

Design(设计方案)

在各算子的 op_host 侧完成 ascend960dt 适配:

  1. CMakeLists.txt:新增 ascend960dt 编译分支,COMPUTE_UNIT ascend960dt_968B7,选项 --cce-auto-sync=off -mllvm -cce-aicore-dcci-before-kernel-end=false -Wno-deprecated-declarations -Werror。
  2. *_def.cpp:新增 AddConfig("ascend960dt", regbaseCfg)(原 config_950 重命名为 regbaseCfg)。
  3. 新增 config/ascend960dt/ 下的 <op>_binary.json 与 <op>_simplified_key.ini(CausalConv1d、MaskedCausalConv1d、MaskedCausalConv1dBackward)。

关联 PR:https://gitcode.com/cann/ops-transformer/pull/12440

likedislike
YYanWenkai
15 天前 添加了label:requirement
YanWenkai
15 天前 评论:

/assign @wkyan

likedislike
CANN-robotCANN-robot成员
15 天前 将 wkyan 设为负责人
YYanWenkai
15 天前 修改标题为 “[Requirement|需求建议]: causal_conv1d 系列算子适配 ascend960平台”,原标题为“[Requirement|需求建议]: causal_conv1d 系列算子适配 ascend960dt 平台”
YYanWenkai
15 天前 修改了issue 的描述
CANN-robotCANN-robot成员
14 天前 关闭了 issue
CANN-robotCANN-robot成员
14 天前 添加了label:resolved