近日,CMU 助理教授贾志豪(Zhihao Jia)团队创新玩法,推出了一个名为「Mirage Persistent Kernel(MPK)」的编译器,能够自动将 LLM 转化为优化的巨型内核(megakernel),从而将 LLM 推理延长降低 1.2 到 6.7 倍。
在这种设计中,系统仅启动一个 GPU 内核来执行整个模型 —— 从逐层推算到 GPU 间通讯 —— 整个过程无需中断。这种步骤提供了以下几个关键的机能优势:
解除内核启动开销:通过预防沉复的内核挪用,即便是在多 GPU 环境下,也能解除内核启动开销;实现跨层软件 pipeline 允许内核在推算当前层的同时,起头为下一层加载数据;沉叠推算与通讯:由于巨型内核能够同时执行推算操作和 GPU 间通讯,从而暗藏通讯延长。
现有的高级 ML 框架 —— 如 PyTorch、Triton 和 TVM,它们自身并不支持端到端巨型内核天生。此表,现代 LLM 系统由各类分歧的专用内核库构建而成:用于通讯的 NCCL 或 NVSHMEM,用于高效把稳力推算的 FlashInfer 或 FlashAttention,以及用于自界说推算的 CUDA 或 Triton。
那么能否通过编译自动化这个过程呢?受到这个问题的启发,来自 CMU、华盛顿大学、加州大学伯克利分校、英伟达和清华大学的团队开发出了 MPK—— 一个编译器和运行时系统,它能自动将多 GPU 的 LLM 推理转换为高机能的巨型内核。MPK 开释了端到端 GPU 融合的效力优势,同时只必要开发者支出极幼的手动致力。
MPK 的一个关键优势在于:通过解除内核启动开销,并最大水平地沉叠跨层的推算、数据加载和 GPU 间通讯,实现了极低的 LLM 推理延长。
除了单 GPU 优化,MPK 还将推算与 GPU 间通讯融合进一个单一的巨型内核。 这种设计使得 MPK 可能最大水平地沉叠推算与通讯。因而,MPK 相对于当前系统的机能提升随着 GPU 数量的增长而增大,使其在多 GPU 部署场景下尤为高效。
Part 1:MPK 编译器,其将 LLM 的推算图转化为优化的工作图;Part 2:MPK 运行时系统,该系统在单个巨型内核内执行工作图,以实现高吞吐量与低延长。
LLM 的推算过程通常暗示为推算图,其中每个节点对应一个推算算子(如矩阵乘法、把稳力机造)或集中通讯原语(如 all-reduce),边暗示算子间的数据依赖关系。现有系统通常为每个算子启动独立的 GPU 内核。
然而,这种「单算子单内核」的执行模型难以实现 pipeline 优化,由于依赖关系是在整个内核的粗粒度层面强造执行的,而非现实数据单元层面。
典型案例如矩阵乘法(matmul)后接 all-reduce 操作:现有系统中,all-reduce 内核必须期待整个 matmul 内核实现。而现实上,all-reduce 的每个数据分块仅依赖 matmul 输出的部门了局。这种逻辑依赖与现实依赖的错配,严沉限度了推算与通讯的沉叠潜力。
下图 2 展示了 MPK 编译器将 PyTorch 界说的 LLM 推算图转化为优化细粒度工作图,最大化露出并行性。右侧展示次优规划 —— 其引入不用要的数据依赖与全局樊篱,导致跨层流水线优化机遇受限。
为相识决此问题,MPK 引入的编译器可将 LLM 推算图自动转化为细粒度工作图。该工作图在子内核级别显式捕获依赖关系,实现更激进的跨层流水线优化。
工作(矩形暗示),代表分配给单个 GPU 流式多处置器(SM)的推算 / 通讯单元。事务(圆形暗示),暗示工作间的同步点。触发机造,每个工作发出指向触发事务的边,该事务在关联工作全数实现后激活。依赖机造,每个工作接管来自依赖事务的边,批注事务激活后工作立即启动。
工作图使 MPK 可能挖掘推算图中无法实现的 pipeline 优化机遇。例如,MPK 能够构建优化工作图 —— 其中每个 all-reduce 工作仅依赖于天生其输入的对应 matmul 工作,从而实现分块执行与推算通讯沉叠。
MPK 蕴含内置 GPU 运行时系统,可在单个 GPU 巨型内核内齐全执行工作图。这使得系统能在推理过程中无需额表内核启动的情况下,实现工作执行与调度的细粒度节造。
获取工作:从队列中提取下一待执行工作。执行推算:运行工作(如矩阵乘法 / 把稳力机造 / GPU 间数据传输)。事务触发:工作实现后通知触发事务。循环执行:沉复上述过程。
调度决策由 MPK 的散布式调度单元处置,每个调度单元运行于单个线程束(warp)上。由于每个流式多处置器(SM)能够包容多个线程束,因而单 SM 最多可并发运行 4 个调度单元。每个调度单元守护激活事务队列,并持续执行以下操作:
下图 3 展示了 MPK 的执行功夫线,其中每个矩形代表一个在工作单元上运行的工作;每个圆圈代表一个事务。当一个工作实现时,它会递增其对应触发事务的计数器。当事务计数器达到预设阈值时,该事务被视为已激活,并被参与调度单元的事务队列。随后,调度单元会启动所有依赖于该事务的下游工作。
由于所有的调度和工作切换都产生在单一内核高低文内,工作间的开销极低,通常仅需 1-2 微秒,从而可能高效地执行多层、多 GPU 的 LLM 工作负载。
团队对 MPK 的愿景是使巨型内核编译既易于使用又具备高机能。目前,你只需几十行 Python 代码(重要用于指定巨型内核的输入和输出)即可将一个 LLM 编译成一个巨型内核。此方向仍有辽阔的索求空间,目前在积极攻关的一些关键领域蕴含如下:
支持现代 GPU 架构。下一个里程碑是将 MPK 扩大到支持下一代架构,例如 NVIDIA Blackwell。一个重要挑战在于若何将线程束专业化,这是新型 GPU 的一项关键优化技术,与 MPK 的巨型内核执行模型相集成。处置工作负载动态性。MPK 目前构建的是静态工作图,这限度了它处置动态工作负载(如 MoE 模型)的能力。团队在开发新的编译战术,使 MPK 可能在巨型内核内部支持动态节造流和前提执行。高级调杜纂工作分配。MPK 在职务级别解锁了新的细粒度调度能力。固然当前的实现使用单一的轮询调度在流式多处置器(SM)之间分配工作,但团队看到了在高级调度战术(如优先级感知或吞吐量优化战术)方面令人兴奋的机遇,可利用于诸如延长服务等级指标(SLO)驱动的服务或混合批处置等场景。
团队相信,MPK 代表了在 GPU 上编译和执行 LLM 推理工作负载方式的底子性转变,并热切等待与社区合作,共同推动这一愿景向前发展。
激战雪姨1到40章的作者唐枫是谁当今足坛,莫德里奇这样的低调奢华大师凤毛麟角。2002年出道的莫德里奇,在皇马13年拿过28座冠军奖杯,是皇马历史上冠军奖杯最多的球员。在克罗地亚国家队,魔笛一直都是真核。踢完2026年世界杯,莫德里奇可能选择退役,结束辉煌的足球生涯。魔笛的表演,看一场少一场。5月25日的美职联,迈阿密国际VS费城联合。比赛第70分钟,梅西主动要求离场,疑似受伤。赛后的检查结果显示,左腿腘绳肌因肌肉疲劳出现负荷劳损。梅西的情况,让外界感到担忧,世界杯在即,阿根廷球迷希望梅西无大碍。激战雪姨1到40章的作者唐枫是谁〖费视频》我一直说,在足球世界里,在体育世界里,不可能永远赢球。上赛季我们没有达到预期,但我们已经找到了问题所在。球队因为世俱杯没能完成正常的季前备战,这影响了整个赛季。赛季前半段我们遭遇了将近30次伤病,其中一些还是长期伤病,这些因素起到了决定性作用。如今,恩里克是足坛最炙手可热的教练。随着他率巴黎圣日耳曼夺得个人第二座欧冠奖杯,再加上瓜迪奥拉将告别曼城的消息传出,这位阿斯图里亚斯教头已经成为世界足坛教练席上的标杆。没有人比“卢乔”更能代表一名成功、赢家型主帅的形象;同时,他还是一名骨子里的巴萨人,也公开表明过自己的反皇马立场。
20260607 ? 激战雪姨1到40章的作者唐枫是谁他承认自己怀疑这个项目涉嫌巨大风险,也承认模式必须靠“源源不断的新人”才能维持。他甚至给记者画出了资金流向图:“我们投的3万块,等于充公到前面星级领导手里了。我赚的钱,是你比我晚投的钱返给我的。”《错轨(N)作者:彭彭的幼说》《萨卡:此刻正当时》把两代阿森纳名宿联系在一起。片中,萨卡与俱乐部传奇亨利同框,制作方称这将是不设防地呈现顶级球员如何在最高水平舞台上表现。萨卡和亨利谈到了压力、韧性,以及最初吸引他们走上足球之路的快乐。这场对话也把他们带回萨卡位于伊灵的童年住所,一切正是从那里开始。
20260607 ? 激战雪姨1到40章的作者唐枫是谁科技“旧势力”回归的背后,是AI产业逻辑演进的“新阶段”。受访的机构人士指出,随着AI基建从单纯“买GPU”迈入“买一切配套”的全面扩容期,手握成熟制造与交付能力的硬件厂商,正成为AI物理世界的关键“建造者”。相关标的的“稀缺溢价”,也正在得到重估。《叔叔的娇娃娃LN笔趣阁最新章节列表》回到裁判问题,随着法丙联赛的创建,将实施新的机制,特别是视频挑战,每场比赛每位教练可使用两次。这是否旨在提高对裁判决定的理解,加强裁判教育,并减少对裁判的制裁感或压力?