公共定义
定义
1 2 3 4 5 6 7 8 9 | enum ExecuteType : int { EXECUTE_NORMAL = 0, EXECUTE_PRELAUNCH, EXECUTE_LAUNCH, }; enum LaunchMode : int { KERNEL_LAUNCH_MODE = 0, GRAPH_LAUNCH_MODE }; |
参数列表
下发接口形式枚举,通过Context选择加速库算子下发接口的形式, 支持单段下发和使用分线程两段式下发。
- ExecuteType
成员
描述
EXECUTE_NORMAL
直接下发。
EXECUTE_PRELAUNCH
用于分线程下发,第一段下发。
EXECUTE_LAUNCH
用于分线程下发,第二段下发。
算子下发模式枚举,通过Context选择算子下发的模式,支持单算子下发与整图下发。
- LaunchMode
成员
描述
KERNEL_LAUNCH_MODE
单算子下发模式。
GRAPH_LAUNCH_MODE
整图下发模式。
父主题: atb/context.h