CA88

EN CA88(中国区)唯一官方网站 CA88(中国区)唯一官方网站
www.ahsjsjt.cn

《对面邻居不拉窗帘》作者:小花灯糕舍弃CUDA编程!CMU蹬酌代码将LLM编译成巨型内核,推理延长降6.7倍

近日,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 推理工作负载方式的底子性转变,并热切等待与社区合作,共同推动这一愿景向前发展。

《对面邻居不拉窗帘》作者:小花灯糕
《对面邻居不拉窗帘》作者:小花灯糕新华社北京6月3日电(记者陈钟昊)6月2日至3日,由国务院新闻办公室指导的“2026世界市长对话·北京”活动在北京市举行。来自摩洛哥、塞舌尔、突尼斯、塞内加尔等6个非洲国家的30余位市长和城市代表,以及世界旅游城市联合会的部分会员代表共同参加了活动。与会嘉宾围绕“加强文明交流互鉴 赓续中非传统友好”开展深入交流与互动,共同探索中非人文交流与城市合作的发展之路。2026年世界杯不仅是竞技与荣耀的舞台,也充满了各种有趣的巧合与数据亮点。其中一项引人关注的趣闻是:本届赛事共有149名球员将在比赛期间迎来自己的生日。《对面邻居不拉窗帘》作者:小花灯糕〖费行情.WWW百度全行搜索》事实上,日本2025财年的防卫预算就已提前达到GDP 2%的目标,而2026财年的防卫预算更是突破9万亿日元大关。相比之下,中国作为世界第二大经济体,国防预算占GDP比重常年维持在1.5%以内,始终低于世界平均水平。日本在自身军费连年飙升、大规模采购进攻性武器的同时,却对中国的正常国防建设指手画脚。这种“贼喊捉贼”的把戏,暴露了其蓄意破坏地区战略平衡的真实图谋。王恒的画室里,依旧满墙满地都是他以双臂为手作的画。他说,读研最大的收获是,花鸟画写生、创作有了明显的进步,对艺术的理解也更深了。他的硕士毕业作品《凌云》获评为优秀毕业作品。
20260605 ? 《对面邻居不拉窗帘》作者:小花灯糕本届皇马主席选举给皇马球迷以及普通足球球迷都留下了不少名场面。为了继续制造传播热度,弗洛伦蒂诺-佩雷斯的竞选团队决定用一段视频展示穆里尼奥。如果弗洛伦蒂诺赢得选举,穆里尼奥将成为皇马主教练。汉子与女人高清免费版电视剧这一套组合拳,诚意拉满。而密歇根,这个曾用汽车工业撑起美国工业心脏的老牌重镇,正被OpenAI用1GW的电,重新点亮成AI时代的新核心。
《对面邻居不拉窗帘》作者:小花灯糕
? 宣照亮记者 吴俣 摄
20260605 ? 《对面邻居不拉窗帘》作者:小花灯糕该项目旨在将人工智能技术与美国国家实验室所拥有的海量科学数据及超算能力相结合。通过大幅缩短实验和复杂计算所需的时间,力求催生大量具有突破性的科学与技术成果。该项目的覆盖范围横跨26个不同领域,包括半导体研发和关键矿产资源勘探。《《部长出差的日子》在哪里看》农业现代化是国家现代化的基石。“十五五”规划明确提出要加快农业关键核心技术攻关,推动数字技术与农业深度融合。牧原与阿里云的实践,正是“人工智能+现代农业”的典型范例。
《对面邻居不拉窗帘》作者:小花灯糕
? 俞蓉记者 吴虹云 摄
? “压力”是杨妍反复提到的词。刚入行时,她对幼教充满美好憧憬——亲和、温柔、有感染力。但三年下来,理想被现实一点点磨灭:工作强度大、回不完的信息、处理不完的矛盾、还要在监控下工作……杨妍停顿了一下,“有时候甚至开始质疑曾经的选择。”罪过之渊动漫全集
扫一扫在手机打开当前页
【网站地图】