夜雨聆风学习资料网

ARTICLE · 1339898

拆解 TensorRT-LLM 源码:C++ 物理内存池与

拆解 TensorRT-LLM 源码:C++ 物理内存池与

拆解 TensorRT-LLM 源码:C++ 物理内存池与

跑 2048×2048、60 帧视频,DiT 的时空 Token 会从 24.5 万飙到 98.3 万。单卡 24GB,没优化的话,第一步去噪就爆显存。

很多人以为开了 FlashAttention-2 就稳了。但在百万级序列面前,QKV 激活值、LayerNorm 特征,加上 T5、CLIP、VAE 常驻显存,24GB 根本不够用。再切线上多风格 LoRA,静态合并还会卡请求、造碎片。

物理内存池管分配:它不走传统动态申请,而是虚拟页映射到连续物理内存。运行时几乎零碎片,释放后微秒级复用,长文本不再被碎片顶出去。

CuTe 管计算:密集注意力会暴涨显存。CuTe 按细粒度二维布局只加载非零块,配合原地计算,显存能砍掉七成以上。

代码里还接了 CUDA 异步流做换入,以及微秒级打桩的动态 LoRA 融合。说白了,FlashAttention 只是润滑剂,真要扛百万 Token,得从内存布局一路改到算子调度。

原文拆解 TensorRT-LLM 源码:C++ 物理内存池与 CuTe 稀疏算子如何切开 DiT 显存?
作者提示: 个人观点,仅供参考
广东,37分钟前,

回首页,搜索你想要的文档

在首页搜索框输入关键词,快速定位资料

文库首页搜索指引 立即搜索

相关学习资料