★ 2
ATRL：面向昇腾 Triton 内核生成的压缩感知强化学习
论文提出 ATRL（Ascend Triton Reinforcement Learning）后训练方法，将 GRPO 改造为压缩感知的分段级目标，用于长程内核生成任务。方法配套轻量智能体框架，提供编译、正确性与性能剖析的级联反馈，并注入昇腾专用知识库。在昇腾硬件上的 NPUKernelBench…
arXiv cs.LG13 天前
#Triton 内核
共 2 篇相关资讯，按重要度与时间排序，每 3 小时更新。
★ 2
PolyGraph：面向自适应投机解码的计划缓冲 CUDA Graph
论文提出 PolyGraph，用计划缓冲机制让自定义 Triton 内核只捕获一张最大网格 CUDA Graph，每次重放前把活跃形状写入 GPU 显存。在单张英伟达 H800 上，以 Qwen2.5-3B 维度做原型，将 64 个形状专用图合并为 1 个，P50 步延迟与基线持平，图构建时间从 …
arXiv cs.LG13 天前
这个主题的相关资产
相关公司英伟达（Nvidia）1