causal_conv1d 系列算子(CausalConv1d、FusedCausalConv1d、InplaceFusedCausalConv1d、MaskedCausalConv1d、MaskedCausalConv1dBackward)此前仅配置了 ascend950 / ascend350 等平台,尚未适配 ascend960平台,需要补齐该平台的算子定义与编译配置。
使 causal_conv1d 系列算子可在 ascend960dt 平台上编译与运行,扩展该系列算子的芯片支持范围。
在各算子的 op_host 侧完成 ascend960dt 适配:
CMakeLists.txt
ascend960dt
COMPUTE_UNIT ascend960dt_968B7
--cce-auto-sync=off -mllvm -cce-aicore-dcci-before-kernel-end=false -Wno-deprecated-declarations -Werror
*_def.cpp
AddConfig("ascend960dt", regbaseCfg)
config_950
regbaseCfg
config/ascend960dt/
<op>_binary.json
<op>_simplified_key.ini
关联 PR:https://gitcode.com/cann/ops-transformer/pull/12440
/assign @wkyan
Backgroud(背景信息)
causal_conv1d 系列算子(CausalConv1d、FusedCausalConv1d、InplaceFusedCausalConv1d、MaskedCausalConv1d、MaskedCausalConv1dBackward)此前仅配置了 ascend950 / ascend350 等平台,尚未适配 ascend960平台,需要补齐该平台的算子定义与编译配置。
Benefit / Necessity(价值/作用)
使 causal_conv1d 系列算子可在 ascend960dt 平台上编译与运行,扩展该系列算子的芯片支持范围。
Design(设计方案)
在各算子的 op_host 侧完成 ascend960dt 适配:
CMakeLists.txt:新增ascend960dt编译分支,COMPUTE_UNIT ascend960dt_968B7,选项--cce-auto-sync=off -mllvm -cce-aicore-dcci-before-kernel-end=false -Wno-deprecated-declarations -Werror。*_def.cpp:新增AddConfig("ascend960dt", regbaseCfg)(原config_950重命名为regbaseCfg)。config/ascend960dt/下的<op>_binary.json与<op>_simplified_key.ini(CausalConv1d、MaskedCausalConv1d、MaskedCausalConv1dBackward)。关联 PR:https://gitcode.com/cann/ops-transformer/pull/12440