Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
为 conversion 目录下的 split_v、split、unpack 三个算子新增 ascend960dt 平台支持,使其能够在 ascend960dt 平台完成算子定义注册与 tiling 编译。
扩展上述算子可运行的硬件平台,使 ascend960dt 平台下可通过 aclnn 等方式调用 split_v、split、unpack 算子。
在算子工程 CMakeLists.txt 中增加 COMPUTE_UNIT 与 TILING_DIR,在 op_host 的 *_def.cpp 中 AddConfig,补充 op_host/config/ascend960dt 下的 binary config 与 simplified_key,并在 scripts/kernel/binary_config/ascendc_config.json 中登记 ascend960dt 的编译配置,使算子编出 ascend960dt 版本。
this->AICore().AddConfig("ascend960dt", aicoreConfig);
op_host/config/ascend960dt/
split_binary.json
split_v_binary.json
unpack_binary.json
*_simplified_key.ini
scripts/kernel/binary_config/ascendc_config.json
compile_options "ascend960dt": ["-mllvm -cce-aicore-dcci-before-kernel-end=false"]
新增 ascend960dt;原有支持平台保持不变(split_v/unpack:ascend310p、ascend910_93、ascend910b、ascend950、ascend350、mc62;split:ascend950、mc62、ascend350)。
本次改动未修改算子对外约束与数据类型限制,仅新增 ascend960dt 平台支持。
💡 备注(选填)
/assign
Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
一、背景信息 (必填)
为 conversion 目录下的 split_v、split、unpack 三个算子新增 ascend960dt 平台支持,使其能够在 ascend960dt 平台完成算子定义注册与 tiling 编译。
二、价值/作用 (必填)
扩展上述算子可运行的硬件平台,使 ascend960dt 平台下可通过 aclnn 等方式调用 split_v、split、unpack 算子。
三、设计方案 (必填)
3.1 使能方式
在算子工程 CMakeLists.txt 中增加 COMPUTE_UNIT 与 TILING_DIR,在 op_host 的 *_def.cpp 中 AddConfig,补充 op_host/config/ascend960dt 下的 binary config 与 simplified_key,并在 scripts/kernel/binary_config/ascendc_config.json 中登记 ascend960dt 的编译配置,使算子编出 ascend960dt 版本。
3.2 总体设计
3.2.1 算子支持的数据类型
3.2.2 host侧设计
this->AICore().AddConfig("ascend960dt", aicoreConfig);。op_host/config/ascend960dt/目录:split_binary.json、split_v_binary.json、unpack_binary.json及对应*_simplified_key.ini。scripts/kernel/binary_config/ascendc_config.json:Split、SplitV、Unpack 的 compute_units 增加 ascend960dt,并增加compile_options "ascend960dt": ["-mllvm -cce-aicore-dcci-before-kernel-end=false"]。3.3 支持硬件
新增 ascend960dt;原有支持平台保持不变(split_v/unpack:ascend310p、ascend910_93、ascend910b、ascend950、ascend350、mc62;split:ascend950、mc62、ascend350)。
3.4 算子约束限制
本次改动未修改算子对外约束与数据类型限制,仅新增 ascend960dt 平台支持。
💡 备注(选填)