CANN/atvoss内核调度运行接口

📅 2026/7/4 22:00:46 👁️ 阅读次数 📝 编程学习
CANN/atvoss内核调度运行接口

BaseKernelSchedule::Run

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

功能说明

kernel层schedule基类,默认调度策略和用户自定义调度策略必须继承自该类,Run基类接口执行调度策略。

所属头文件链接

/include/elewise/kernel/schedule.h

函数原型

template <typename BlockOp, const auto &Policy, typename ScheduleCfg> class BaseKernelSchedule { template <typename OpParam, typename... Args> aicore inline void Run(OpParam& cfg, Args... args) }

参数说明

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

返回值说明

返回值数据类型返回值说明
voidNA

约束说明

NA

使用示例

暂无

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

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