Scheduler 层扩展开发指南【免费下载链接】cannbot-skillsCANNBot 是面向 CANN 开发的用于提升开发效率的系列智能体本仓库为其提供可复用的 Skills 模块。项目地址: https://gitcode.com/cann/cannbot-skills适用路径blaze_custom路径 A§1 接口规范构造函数MyBlockScheduler(const ProblemShape shape, const Params params)内部计算mCnt_ CeilDiv(m, baseM)、nCnt_、totalCnt_、round_等状态。必需 6 接口接口职责GetTileNum()当前核需处理的 tile 轮数GetTileIdx(BlockCoord)按轮次返回 tile 坐标false 结束GetBlockShape(BlockCoord)tile 实际 M/N 尺寸含尾块修正GetTileL1Shape(mL1, nL1, kL1)L1 级 tile 尺寸GetTileL0Shape(mL0, nL0)L0 级 tile 尺寸GetBlockCoord通过GetTileIdx输出 (mIdx, nIdx)Params 结构体struct Params { int64_t baseM, baseN; int64_t mTailTile, nTailTile; // 尾块拆分数默认 1 int64_t mBaseTailSplitCnt, nBaseTailSplitCnt; int64_t mTailMain, nTailMain; };§2 蛇形遍历模板Z-scan 遍历 M-N 网格奇数行反转 N 方向提升 L2 cache 复用bool GetTileIdx(BlockCoord blockCoord) { if (roundIdx_ round_) return false; int64_t tileIdx blockIdx_ roundIdx_ * blockNum_; int64_t rowIdx tileIdx / nCnt_ / mCoreNum_; int64_t mTileIdx, nTileIdx; if (rowIdx mainRow_) { mTileIdx rowIdx * mCoreNum_ tileIdx % mCoreNum_; nTileIdx (tileIdx / mCoreNum_) % nCnt_; } else { int64_t tailIdx tileIdx - mainRow_ * mCoreNum_ * nCnt_; mTileIdx mainRow_ * mCoreNum_ tailIdx % mTailCoreNum_; nTileIdx (tailIdx / mTailCoreNum_) % nCnt_; } if (rowIdx 1) nTileIdx nCnt_ - 1 - nTileIdx; GetMNK_M(blockCoord) mTileIdx * baseM_; GetMNK_N(blockCoord) nTileIdx * baseN_; roundIdx_; return true; }WINDOW_LEN 4控制蛇形行宽mCoreNum_ Min(4, mCnt_)。§3 尾块处理模式策略实现适用尾块合并mBaseTailSplitCnt 1尾块 baseM/2尾块拆分mTailTile 1尾块较大且核数有余尾块独立默认mTailTile1通用默认尾块尺寸mBaseTail_ m - (mCnt_ - 1) * baseM_。GetBlockShape按mTileIdx mBaseNormCnt_判断尾块区域并修正。§4 Group Scheduler 扩展GroupMatmulBlockSchedulerSplitM支持跨 group 迭代void UpdateNextProblem(const TupleShape problemShape) { m_ GetMNK_M(problemShape); n_ GetMNK_N(problemShape); mCnt_ CeilDiv(m_, baseM_); nCnt_ CeilDiv(n_, baseN_); totalCnt_ mCnt_ * nCnt_; roundIdx_ 0; // 重置轮次 } void UpdateBaseM(uint32_t baseM) { baseM_ baseM; }Kernel 层外层遍历 group内层GetTileIdx遍历当前 group 的 tile。模块索引 →references/modules/blaze-custom/scheduler-modules.md【免费下载链接】cannbot-skillsCANNBot 是面向 CANN 开发的用于提升开发效率的系列智能体本仓库为其提供可复用的 Skills 模块。项目地址: https://gitcode.com/cann/cannbot-skills创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考
