CANN/atvoss:Kernel调度配置生成API

📅 2026/7/14 5:37:26 👁️ 阅读次数 📝 编程学习
CANN/atvoss:Kernel调度配置生成API

BaseKernelSchedule::MakeScheduleConfig

【免费下载链接】atvossATVOSS(Ascend C Templates for Vector Operator Subroutines)是一套基于Ascend C开发的Vector算子库,致力于为昇腾硬件上的Vector类融合算子提供极简、高效、高性能、高拓展的编程方式。项目地址: https://gitcode.com/cann/atvoss

功能说明

kernel层schedule基类,默认调度策略和用户自定义调度策略必须继承自该类,MakeScheduleConfig基类接口根据传入的参数信息,生成scheduleCfg配置信息。

所属头文件链接

/include/elewise/kernel/schedule.h

函数原型

template <typename BlockOp, const auto &Policy, typename ScheduleCfg> class BaseKernelSchedule { template<typename Args> static bool MakeScheduleConfig(const Args& arguments, ScheduleCfg &kernelParam) }

参数说明

参数名称参数类型输入/输出数据类型参数说明默认值
BlockOp模板参数输入NAblock层对象类型,跟kernel层是被包含关系NA
Policy模板参数输入NAkernel层的用户静态策略类型NA
ScheduleCfg模板参数输出NAkernel层调度配置类型NA
Args模板参数输入NA用户的输入参数列表,类型根据用户传入的参数实例化NA
arguments函数形参输入Args用户的输入参数列表NA
kernelParam函数形参输出ScheduleCfgkernel层schedule配置参数NA

返回值说明

返回值数据类型返回值说明
bool生成scheduleCfg配置信息成功还是失败,true:成功,false:失败

约束说明

NA

使用示例

暂无

【免费下载链接】atvossATVOSS(Ascend C Templates for Vector Operator Subroutines)是一套基于Ascend C开发的Vector算子库,致力于为昇腾硬件上的Vector类融合算子提供极简、高效、高性能、高拓展的编程方式。项目地址: https://gitcode.com/cann/atvoss

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考