返回 文章 apply CMS 文章

NVIDIA 机密计算赋能高性能私有 AI 推理:TensorRT LLM 的 CC 感知优化实践

NVIDIA 通过 TensorRT LLM 的 CC 感知优化,在机密计算环境下实现近乎无损的高性能 AI 推理。

NVIDIA机密计算TensorRT LLMAI 推理
成长分 / 100 76 综合收获、行动、留存与影响

NVIDIA 机密计算赋能高性能私有 AI 推理:TensorRT LLM 的 CC 感知优化实践
为什么值得读了解机密计算如何在不显著牺牲性能的前提下保护敏感数据和专有模型。

获取 TensorRT LLM 针对 CC 环境的具体优化措施,包括内存移动、内核调优和多 GPU 通信。

关键洞察
  1. 机密计算通过内存加密的 CVM、机密 GPU 和加密 NVLink 提供可信执行环境,但会改变内存移动、时序、调度和多 GPU 通信的假设。
  2. 选择长输入上下文、扩展输出生成和低并发度的工作负载可以更明显地暴露 CC 开销。
  3. TensorRT LLM 通过 CC 感知的内存选择、异步回读、使用 GPU %globaltimer 替代 CUDA events 进行自动调优,以及检测 NVLS 可用性来减少性能开销。
转成行动

深入阅读

正文与原文对照

原文保真覆盖:全文原文字符:9711

随着大语言模型(LLM)推理越来越多地在个人、企业和受监管场景中处理敏感信息和专有模型上下文,数据必须在可信环境内处理。NVIDIA 机密计算(Confidential Computing,CC)借助内存加密的机密虚拟机(CVM)、机密 GPU 和加密的 NVIDIA NVLink,为安全运行这些工作负载提供了一条路径。这使得在可信硬件上运行生产级 AI 推理成为可能。

NVIDIA TensorRT LLM 等推理框架通过将框架层面的优化与 NVIDIA 加速计算相结合,提供同类最佳的 AI 推理。然而,当这些框架运行在启用 CC 的环境中时,安全执行会改变内存搬移、时序、调度和多 GPU 通信背后的假设。如果运行时不做适配,这些变化会带来性能开销。因此,要保持高性能,推理框架与机密计算环境需要协同优化。

对于正在评估 NVIDIA Blackwell GPU 上机密推理的 AI 平台工程师,本文考察了 TensorRT LLM 等 AI 推理框架为适应安全执行、同时帮助保持推理性能而采用的 CC 感知适配。文中还介绍了一套受控的方法,团队可以用它来量化自己工作负载上的 CC 开销。

选择能暴露 CC 开销的工作负载

工作负载的特征决定了 CC 开销的可见程度。较高的请求量可以让停顿与其他工作重叠,从而摊薄固定的加密成本,使直接影响更难观察。

为了暴露这些影响,应选择输入上下文长、输出生成长、并发低的工作负载。长上下文会在预填充(prefill)阶段加大数据搬移的压力,长时间生成会在解码(decode)阶段放大每个 token 上微小的 CC 开销,而低并发则限制了在并发请求之间隐藏这些成本的机会。

NVIDIA 性能工程团队针对此处评估的工作负载采用了这些特征。

表 1. CC 开启与 CC 关闭对比的工作负载配置

参数 配置
模型 nvidia/DeepSeek-R1-0528-NVFP4
推理框架 TensorRT LLM,PyTorch 后端
输入/输出序列长度 32K 输入/1K 输出
并发请求数 1、2、4、8 和 16
并行方式 TP=8, EP=1, PP=1
KV 缓存 FP8

通过受控的 CC 开启与 CC 关闭对比来测量 CC 开销

要分离出 CC 对性能的影响,需要在两种条件下运行同一工作负载:机密计算关闭(CC off)和机密计算开启(CC on),同时保持模型、硬件、框架版本、序列长度、并行方式和并发数不变,使 CC 状态成为唯一变化的变量。

在每个并发级别上:

  • 保留的输出吞吐量:100 x(CC on 输出 token/秒 ÷ CC off 输出 token/秒)
  • 延迟开销,即每输出 token 时间(Time Per Output Token,TPOT):100 x(CC on TPOT ÷ CC off TPOT − 1)

性能团队可以借助这些测量结果,量化针对目标工作负载启用 CC 后保留了多少 CC off 基线性能。NVIDIA 性能工程团队使用表 2 中汇总的硬件和软件配置进行了这一对比。

表 2. CC on 和 CC off 两组运行使用的硬件和软件配置

组件 版本/详情
硬件 1 台 NVIDIA DGX B200 系统(8 块 NVIDIA B200 GPU)
平台 Intel TDX
宿主机操作系统 Ubuntu 25.10
宿主机内核 6.17.0-20-generic
客户机操作系统 Ubuntu 24.04.4 LTS
客户机内核 6.8.0-124-generic
客户机 vCPU 256
客户机 NUMA 2 个节点
NVIDIA 驱动 595.71.05
VBIOS FW 1.4.x [97.10.64.00.0C]
GPU 功耗上限 1,000 W
CUDA 13.2
TensorRT LLM nvcr.io/nvidia/tensorrt-llm/release:1.3.0rc22
NCCL v2.30
OpenSSL 3.6.0
编排 Docker 容器 + NVIDIA Container Toolkit

性能结果

如图 1 和图 2 所示,在并发 1 至 16 范围内,CC on 保留了 CC off 输出 token 吞吐量的 96.1% 至 98.2%,而平均 TPOT 与基线的差距保持在 1.2% 至 4.3% 之内。

水平成对条形图,对比并发为 1、2、4、8 和 16 时 CC off 与 CC on 的输出 token 吞吐量。CC off 归一化为 100%,以深灰色表示;CC on 以绿色表示,分别保留了 98.2%、98.0%、96.1%、96.8% 和 96.8%。
图 1. 并发 1 至 16 下,CC on 输出 token 吞吐量相对于 CC off 基线的比例。CC on 保留了基线吞吐量的 96.1% 至 98.2%
水平成对条形图,对比并发为 1、2、4、8 和 16 时 CC off 与 CC on 的平均 TPOT。CC off 归一化为 100%,以深灰色表示。CC on 以 NVIDIA 绿色表示,分别为 101.2%、103.2%、104.3%、103.5% 和 103.1%。实测的 CC on TPOT 值分别为 3.225、4.571、6.658、10.759 和 18.039 毫秒。数值越低,延迟越好。
图 2. 并发 1 至 16 下,启用 CC 时的平均 TPOT 相对于 CC off 基线的比例。CC on 带来了 1.2% 至 4.3% 的 TPOT 开销(TPOT 越低越好)

识别并降低 CC 开销

NVIDIA Blackwell 机密计算架构引入了由硬件强制执行的安全路径,用于保护使用中的数据和工作负载。有关该架构的详细概述,请参阅《Hardware-Rooted AI Security That Won’t Slow You Down》。

对于 TensorRT LLM 用户和框架开发者,以下细节说明了这些安全路径如何改变常见的运行时假设,以及 TensorRT LLM 如何适配以降低由此产生的性能开销。

适配主机到设备的数据搬移

在 B200 CC 中,由于 GPU 无法直接访问受保护的 CVM 内存,主机到设备的传输需要经过一个由软件加密的中转缓冲区(bounce buffer)。这改变了推理框架的预期行为:锁页内存(pinned memory)不再具有通常的异步传输优势,而且某些拷贝会阻塞调用线程。

  • 主机到设备的缓解措施:TensorRT LLM 采用 CC 感知的内存选择,在受影响的路径上选择可分页内存(pageable memory),而不是无条件使用锁页内存。
  • 设备到主机的缓解措施:TensorRT LLM 将重复的 token 和采样数据回读移到异步工作线程中,避免受保护的拷贝在解码期间阻塞主调度器。详情请参阅 TensorRT LLM PR #11573。

稳定内核自动调优器的计时

内核自动调优器通常使用 CUDA 事件来比较候选策略(tactic)。在测试的 CC 配置中,CUDA 事件时间戳产生的计时信号不稳定,这可能导致自动调优器选择较慢的策略。

  • 缓解措施:TensorRT LLM 在 CC 下使用 GPU %globaltimer 来测量策略,而在 CC 之外仍保留 CUDA 事件。详情请参阅 TensorRT LLM PR #11657。

选择 CC 感知的多 GPU 通信

B200 CC 配置中不提供 NVLS(NVLink SHARP)多播。没有 NVLS,NCCL_SYMMETRIC 就无法发挥其预期的多播收益,却仍可能在使用非多播集合通信路径之前产生内存注册和跨 rank 同步的成本。

  • 缓解措施:面向 CC 的框架应检测 NVLS 是否可用,并针对给定的消息大小、拓扑和工作负载特征,选择使延迟最小的通信算法。

开始使用 NVIDIA 机密计算

NVIDIA 机密计算将硬件强制的保护扩展到机密虚拟机、NVIDIA Blackwell GPU 和加密的 NVLink,在专有模型、企业上下文和敏感提示词被处理期间对其加以保护。

机密计算并不会消除性能工程的需求,反而使框架层面的感知更为重要。有了 TensorRT LLM 针对安全数据搬移、自动调优和多 GPU 通信的 CC 感知适配,在八块 NVIDIA B200 GPU 上进行机密 DeepSeek-R1 推理时,保留了超过 96% 的 CC off 输出 token 吞吐量,同时每 token 延迟开销低于 5%。

随着企业将私有推理推向生产,应将安全配置和推理优化视为同一个部署问题。启用机密计算,对环境进行证明(attest),并使用你打算实际提供服务的工作负载对 CC on 和 CC off 进行基准测试。对于将私有推理推向生产的 AI 平台工程师和 TensorRT LLM 用户,应把安全配置和推理优化视为一项全栈工程工作。

要开始规划你的机密推理部署,请参阅 NVIDIA Trusted Computing 文档,并了解最新的 TensorRT LLM 特性和发布说明。要了解最新进展,请关注 NVIDIA Confidential Compute 新闻。

致谢

感谢 Dan Hansen、Sheel Pethe、Samuel Mendoza-Jonas、Moein Ghaniyoun、Vidhya Krishnan、Avinash Ahuja、Laikh Tewari、Laura Martinez 和 Matheen Raza 在整个工作过程中的工程贡献、技术指导、分析和细致审阅。