Task 模块架构【免费下载链接】runtime本项目提供CANN运行时组件和维测功能组件。项目地址: https://gitcode.com/cann/runtime1. 模块概述功能介绍Task 模块负责管理各类任务的创建、执行和回收。支持多种任务类型核函数执行、内存操作、事件同步、模型执行等通过 TaskInfo 结构体承载任务信息。Stream::taskResMang_ 和 TaskFatory 共同负责任务对象分配和回收。设计目标提供统一的任务管理接口支持多种任务类型扩展实现高效的任务分配和回收机制支持任务状态管理和错误处理2. 使用场景与对外接口2.1 使用场景示意场景一内核执行任务// 下发单算子任务 aclrtLaunchKernel(funcHandle, numBlocks, argsData, argsSize, stream); aclrtLaunchKernelXXX();场景二事件同步任务// 下发event 同步任务并流同步结果 aclrtRecordEvent(event, stream1); // 在 stream1 记录事件 aclrtStreamWaitEvent(stream2, event); // stream2 等待事件 aclrtSynchronizeStream(stream1); // stream1 流同步触发回收 aclrtSynchronizeStream(stream2); // stream2 流同步触发回收2.2 任务类型任务分类任务类型头文件说明计算类任务TS_TASK_TYPE_KERNEL_AICORETS_TASK_TYPE_KERNEL_AICPUTS_TASK_TYPE_KERNEL_AIVECdavinci_kernel_task.h算子核执行内存类任务TS_TASK_TYPE_MEMCPYmemory_task.hH2D、D2D、D2H等异步拷贝任务执行事件同步类任务TS_TASK_TYPE_EVENT_RECORDTS_TASK_TYPE_STREAM_WAIT_EVENTTS_TASK_TYPE_NOTIFY_RECORDTS_TASK_TYPE_NOTIFY_WAITTS_TASK_TYPE_DAVID_EVENT_*TS_TASK_TYPE_MEM_WRITE/WAIT_VALUEevent_task.hnotify_task.hcommon_task.h流间同步、卡间同步等事件等待执行条件类任务TS_TASK_TYPE_LABEL_*TS_TASK_TYPE_STREAM_SWITCHTS_TASK_TYPE_STREAM_ACTIVEcond_op_label_task.h标签控制类管理类任务TS_TASK_TYPE_MAINTENANCETS_TASK_TYPE_MODEL_MAINTAINCETS_TASK_TYPE_MODEL_EXECUTEmaintenance_task.hmodel_maintaince_task.hmodel_execute_task.h维护操作流/事件销毁模型绑定/解绑流模型执行3. 架构总览整体设计思路Task 管理整理分为三部分Task 资源管理Task 资源分为两部分Device 粒度的Task资源和Stream粒度的资源。Task 处理函数Task 处理函数主要是针对不同类型的Task进行任务初始化、SQE封装、Task回收后处理、Task 异常后处理等流程进行统一注册管理。Task 下发流程配合API接口通过task资源申请、任务组装、任务下发的等接口组合完成运行时调度。3.1 Task 资源管理架构3.2 Task 处理函数注册架构3.3 David Task下发流程4. 详细设计4.1 核心流程4.1.1 单算子流上 Task 资源申请和下发流程4.1.2 模型流 Task 资源申请和下发流程4.2 核心机制详解4.2.1 TaskInfo 结构设计设计思想TaskInfo 作为任务的统一载体包含任务类型、参数、状态、位域标记等信息。通过 union 存储不同任务类型的专用信息结构。// 文件位置core/inc/task/task_info.hpp typedef struct tagTaskInfoStru { Stream *stream; // 所属流 tsTaskType_t type; // 任务类型枚举 uint32_t errorCode; // 错误码 uint32_t taskSn; // 流水号profiling 用 uint32_t pos; // SQ 位置David 用 uint16_t id; // 任务 ID uint16_t flipNum; // 翻转计数taskId 回卷 uint8_t profEn; // Profiling 启用标志 ... uint8_t serial : 1; // 串行 ID 模式 uint8_t bindFlag : 1; // 模型绑定流标志 uint8_t isCqeNeedConcern : 1; // CQE 需关注 uint8_t isNeedStreamSync : 1; // 需流同步 uint8_t sqeNum : 7; // SQE 数量David 多 SQE ... union { AicTaskInfo aicTaskInfo; AicpuTaskInfo aicpuTaskInfo; DavinciMultiTaskInfo davinciMultiTaskInfo; MemcpyAsyncTaskInfo memcpyAsyncTaskInfo; ... } u; // 各任务类型专用信息 ... } TaskInfo;4.2.2 TaskRes资源分配核心逻辑TaskResManageDavid 核心分配代码Stream上的TaskId需要严格保序从TaskInfo指针申请到sqe组组装下发到硬件整个过程要和TaskId行为一致。因此整个过程有StreamLock进行控制。多 SQE 设计一个任务可能占用多个 SQE如sqeNum2taskInfo.id都指向首位置pos。回收时按sqeNum刷新taskResAHead_。// task_res_da.cc rtError_t TaskResManageDavid::AllocTaskInfoAndPos(uint32_t sqeNum, uint32_t pos, TaskInfo **task, bool needLog) { const uint16_t head taskResAHead_.Value(); uint16_t tail taskResATail_.Value(); const uint32_t taskDesTailIdx (tail sqeNum) % taskPoolNum_; // 满判断考虑环形队列翻转 if (((tail head) ((tail sqeNum) head)) || ((tail head) ((tail sqeNum) taskPoolNum_) (taskDesTailIdx head))) { return RT_ERROR_TASKRES_QUEUE_FULL; } pos tail; *task taskRes_[pos].taskInfo; (void)memset_s(*task, sizeof(TaskInfo), 0, sizeof(TaskInfo)); // 多 SQE 任务所有 sqeNum 共享同一个 id pos while (tail ! taskDesTailIdx) { taskRes_[tail].taskInfo.id pos; tail (tail 1U) % taskPoolNum_; } taskResATail_.Set(tail); // 更新 tail allocNum_; // 累计分配计数用于周期回收触发 return RT_ERROR_NONE; }队列满时的回收重试// task_david.cc rtError_t AllocTaskInfo(TaskInfo **taskInfo, Stream *stm, uint32_t pos, uint32_t sqeNum) { TaskResManageDavid *taskResMang ...; rtError_t error taskResMang-AllocTaskInfoAndPos(sqeNum, pos, taskInfo); while (error RT_ERROR_TASKRES_QUEUE_FULL) { // 释放 StreamLock尝试回收 stm-StreamUnLock(); TryToReclaimTask(stm, needLog); // 调用 TaskReclaimByStream 或唤醒回收线程 stm-StreamLock(); // 重试分配 error taskResMang-AllocTaskInfoAndPos(sqeNum, pos, taskInfo, needLog); } if (error RT_ERROR_NONE) { Runtime::Instance()-AllocTaskSn((*taskInfo)-taskSn); } return error; }4.2.3 Task处理函数的回调框架1. TaskFuncSingle 结构体不同 TASK_TYPE 的任务回调函数结构体针对不同 CHIP_TYPE 和 taskType 单独处理调用函数 rtError_t RegTaskFunc(rtChipType_t chipType, tsTaskType_t taskType, const TaskFuncSingle funcs) 进行全局注册。// task_manager.h:61-70 struct TaskFuncSingle { PfnTaskToCmd toCommandFunc; PfnTaskToSqe toSqeFunc; PfnDoCompleteSucc doCompleteSuccFunc; PfnTaskUnInit taskUnInitFunc; PfnWaitAsyncCpCompleteFunc waitAsyncCpCompleteFunc; PfnPrintErrorInfo printErrorInfoFunc; PfnTaskSetResult setResultFunc; PfnTaskSetStarsResult setStarsResultFunc; };2. TaskFuncArrays 结构体TaskFuncArrays g_taskFuncArrays[CHIP_END] runtime初始化的时候根据 CHIP_TYPE 保存对应 TaskFunc。// task_manager.h struct TaskFuncArrays { PfnTaskToCmd toCommandFunc[TS_TASK_TYPE_RESERVED]; // [0] TaskCommand 构建 PfnTaskToSqe toSqeFunc[TS_TASK_TYPE_RESERVED]; // [1] SQE 构建 PfnDoCompleteSucc doCompleteSuccFunc[TS_TASK_TYPE_RESERVED]; // [2] Task完成后处理 PfnTaskUnInit taskUnInitFunc[TS_TASK_TYPE_RESERVED]; // [3] Task指针清空 PfnWaitAsyncCpCompleteFunc waitAsyncCpCompleteFunc[TS_TASK_TYPE_RESERVED]; // [4] 异步等待 PfnPrintErrorInfo printErrorInfoFunc[TS_TASK_TYPE_RESERVED]; // [5] 错误打印 PfnTaskSetResult setResultFunc[TS_TASK_TYPE_RESERVED]; // [6] Task异常结果设置v100 PfnTaskSetStarsResult setStarsResultFunc[TS_TASK_TYPE_RESERVED]; // [7] Task异常结果设置v100 };3. Task处理时使用的全局变量业务代码中通过 g_XXX 对 taskType 进行回调处理。static PfnTaskToCmd *g_toCommandFunc g_taskFuncArrays[CHIP_BEGIN].toCommandFunc; static PfnTaskToSqe *g_toSqeFunc g_taskFuncArrays[CHIP_BEGIN].toSqeFunc; static PfnDoCompleteSucc *g_doCompleteSuccFunc g_taskFuncArrays[CHIP_BEGIN].doCompleteSuccFunc; static PfnWaitAsyncCpCompleteFunc *g_waitAsyncCpCompleteFunc g_taskFuncArrays[CHIP_BEGIN].waitAsyncCpCompleteFunc; static PfnPrintErrorInfo *g_printErrorInfoFunc g_taskFuncArrays[CHIP_BEGIN].printErrorInfoFunc; static PfnTaskSetResult *g_setResultFunc g_taskFuncArrays[CHIP_BEGIN].setResultFunc; static PfnTaskSetStarsResult *g_setStarsResultFunc g_taskFuncArrays[CHIP_BEGIN].setStarsResultFunc; PfnTaskUnInit *g_taskUnInitFunc g_taskFuncArrays[CHIP_BEGIN].taskUnInitFunc;4.2.4 任务回收流程4.2.4.1 David 任务回收触发时机触发条件调用方式说明每 64 个任务下发后WakeUpRecycleThread周期性回收allocNum_ % 64 0时触发下发任务队列满TryToReclaimTask分配队列满时触发主动回收流同步SynchronizeWakeUpRecycleThread出发回收线程等待所有任务完成流销毁TearDownTaskReclaimByStream清空所有任务独立回收线程RecycleThreadDoForStarsV2IsSeparateSendAndRecycle()模式4.2.4.2 独立回收线程// task_recycle.cc:310-341 void RecycleThreadDoForStarsV2(Device *deviceInfo) { // 遍历设备上所有 stream for (const auto id : streamIdList) { stream-StreamRecycleLock(); stream-ProcArgRecycleList(); // 处理 arg 回收列表 // 跳过空队列 / DVPP 绑定 / 非独立回收模式 if (taskResMang-IsEmpty() || stream-IsBindDvppGrp() || !stream-IsSeparateSendAndRecycle()) { continue; } TaskReclaimForSeparatedStm(stream); // → ProcLogicCqUntilEmpty(stm) // 先收 CQE // → RecycleTaskBySqHeadForRecyleThread(stm) // 再根据 drv sqHead 回收 stream-StreamRecycleUnlock(); } }4.3 模块职责划分模块职责位置TaskInfo任务信息结构体task/inc/task_info.hTaskFactory任务对象工厂类 Device对象持有Task 对象分配和回收task/task_factory.hppTaskAllocator任务对象分配器管理任务池属于TaskFactorytask/task_allocator.hppTaskResManage任务资源管理Stream持有task/task_res_manage/DavidSendTaskSubmitTask任务提交模块task/task_submit/v200/task_david.ccengine/engine.ccRecycleThreadDoForStarsV2任务回收模块task/task_recycle/stars_engine.cc4.4 核心数据结构4.4.1 Device TaskFactory 类图4.4.2 Stream TaskRes 类图5. 关键文件索引| 功能模块 | 文件路径 | 核心内容 | |----------|---------|---------| | Stream 声明 |core/src/stream/stream.hpp|taskResMang_字段 | | DavidStream |core/src/stream/stream_david.cc| CreateStreamTaskResTearDown | | TaskInfo 定义 |core/inc/task/task_info.hpp| TaskInfo 结构体 | | TaskRes 基类 |core/inc/task/task_res.hpp| TaskRes 结构TaskResManage 类声明 | | TaskResManageDavid |core/inc/task/task_res_da.hpp| 原子 head/tailAllocTaskInfoAndPos | | TaskResManage 实现 |core/src/task/task_res_manage/task_res.cc| v100 分配/回收/Load | | TaskResManageDavid 实现 |core/src/task/task_res_manage/v200/task_res_da.cc| David 分配/回收/回滚 | | David 任务下发 |core/src/task/task_submit/v200/task_david.cc| AllocTaskInfoDavidSendTask | | David 任务回收 |core/src/task/task_recycle/v200/task_recycle.cc| TryRecycleTask回收线程 | | 回收公共逻辑 |core/src/task/task_recycle/v200/task_recycle_common_base.cc| TaskReclaimForSeparatedStm回收CQE、回收Task |本模块文档基于源码src/runtime/core/src/task/分析。【免费下载链接】runtime本项目提供CANN运行时组件和维测功能组件。项目地址: https://gitcode.com/cann/runtime创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考