AI 推理术语表

硬件TMA

Tensor Memory Accelerator

先用大白话

NVIDIA 的异步批量张量传输机制,可减少线程逐个计算地址的工作。

技术定义

TMA 是硬件支持的数据传输机制,利用张量描述在受支持的存储空间之间异步搬运数据。

工程细节

Blackwell 文章测试的 global-to-shared 路径可由单个线程发起多维传输。引擎处理地址生成和受支持的布局细节,同步机制通知消费者数据已经就绪。传输形状与在途工作量影响可达带宽。

为什么重要

TMA 能减少部分软件开销,但不保证适合所有访问模式或传输大小。不规则 page 选择或较小传输可能更适合其他 copy 机制。

如何在 InferenceX 中解读

文章在明确的线程、stage 和 box 配置下比较二维 TMA 与 LDGSTS。两者性能交点取决于工作负载,转载也明确标注了原文 bytes 与 KiB 不一致的位置。

在真实基准中理解这一概念

全部文章