三亩地 三亩地SAN MU DI · CODE DIARY
ARTICLE DETAIL

日记详情

真实记录编程学习的某一天,欢迎挑你感兴趣的翻一翻。

CANN/ATVOSS块调度运行接口

CANN/ATVOSS块调度运行接口

BaseBlockSchedule::Run

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

功能说明

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

所属头文件链接

/include/elewise/block/schedule.h

函数原型

template <typename Compute, const auto& Policy, typename ScheduleCfg, typename ArchTagCfg=void> class BaseBlockSchedule { template <typename ArgTup> __aicore__ inline void Run(ScheduleCfg& cfg, ArgTup& argTuple) }

参数说明

参数名称参数类型输入/输出数据类型参数说明默认值
Compute模板参数输入NATile层计算表达图对象类型,跟kernel层是被包含关系NA
Policy模板参数输入NAblock层的用户静态策略类型NA
ScheduleCfg模板参数输入NAblock层调度配置类型void
ArchTagCfg模板参数输入NA用户配置的芯片版本型号NA
ArgTup模板参数输入NA用户的输入参数列表,类型根据用户传入的参数实例化NA
cfg函数形参输入ScheduleCfg用户定义的schedule配置NA
argTuple函数形参输入ArgTup用户的输入参数列表NA

返回值说明

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

约束说明

NA

使用示例

暂无

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

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

← 返回列表