AMD MI355X 上的占用率计算：一份从基本原理出发的指南
本文深入探讨了 AMD MI355X 加速器上的占用率（occupancy）数学原理，从最基本的概念出发，系统性地推导了计算占用率的公式与方法。通过详细分析硬件架构、线程调度与资源限制，为 GPU 性能优化提供了清晰的理论基础和实践指导。
背景速读
- **AMD MI355X** 是 AMD 下一代数据中心 GPU，专为 AI/高性能计算设计，预计 2025-2026 年出货。它是 MI300X 的后继产品，直接对标 NVIDIA 的 B200 "Blackwell" GPU。
- **Occupancy（占用率）** 是 GPU 编程核心概念：指同时活跃的 warp/wavefront（线程束）数量与硬件理论上限之比。高占用率能隐藏内存延迟，但过高可能因寄存器压力反而降低性能。
- 文章从第一性原理推导 MI355X 的占用率计算方式（寄存器、共享内存、wavefront 限制），而非依赖 AMD 官方工具。这类深度分析对优化 AI 推理/训练内核的工程师至关重要，因为 MI355X 的架构细节（如 512KB 寄存器文件、CDNA 4 计算单元设计）尚未被广泛文档化。
- 作者是 GPU 性能优化领域的独立研究者，此前对 MI300X、NVIDIA H100 等做过类似分析。这类"逆向工程"式架构解读在 AI 基础设施社区中价值极高，因为硬件厂商的官方文档常缺少优化所需的底层参数。