<aside>
本文由人类编写
</aside>
基于 Deepseek 新开的论文中所讲,mHC 改进了 HC 的训练过程,且提到新的模型使用了 mHC 进行训练,那么站在推理的角度,针对 mHC 使用的三个融合算子,有必要先用 CuTile 搓一个前向的 demo 出来。用于了解算法内容,以及为后续的调优提供一定的基础。
[!info] arxiv.org
从算子角度看,mHC 的前向计算并不复杂,本质上是:
[!important]
用一个 GEMM + 若干约束函数,生成三组 mixing 系数,再用这些系数去重新组织 residual stream。
为了方便分析,先约定几个维度含义(与论文、实际实现保持一致):
Batch × SeqLenmHC 的扩张系数(通常为 4)hidden sizemHC 首先通过一次大的线性变换,生成所有 mixing 的“原始参数”:
[H̃̃_pre , H̃̃_post , H̃̃_res] = X_t · φ_t