1. PMU在AI芯片中的核心价值
现代AI芯片的电源管理单元(PMU)早已不是简单的电压调节器集合。以我们团队去年流片的AI-29芯片为例,其PMU需要同时应对三个核心挑战:在5W的TDP限制下满足128TOPS的算力需求、支持毫秒级动态电压频率调整(DVFS)、以及实现多电源域的精准协同管理。
这种复杂需求催生了新一代PMU架构的进化。传统PMU可能只需要管理3-5个电源域,而AI-29的PMU需要控制多达23个独立电源域,包括:
- 神经网络处理器(NPU)核心阵列的8个可拆分供电区域
- 4组HBM3内存接口的独立电压调节
- 3种不同工艺节点的混合供电管理(7nm逻辑单元与28nm模拟电路)
实战经验:我们在初期测试时发现,当NPU的8个供电区域同时进行DVFS切换时,会在电源网络上产生高达200mV的纹波。后来通过在PMU中引入交错相位调整技术,将纹波控制在50mV以内。
2. 关键电路设计解析
2.1 自适应降压转换器设计
AI-29采用了基于Sigma-Delta调制的混合型降压转换器,与传统PWM架构相比具有三大创新点:
-
数字辅助模拟控制环:在传统电压模式控制基础上,增加了数字化的纹波预测模块。通过实时监测负载电流变化趋势,提前50ns调整开关管占空比。实测显示,这种方法将瞬态响应时间从300ns缩短到150ns。
-
动态相位分配技术:8相降压转换器可以根据负载情况动态重组:
- 轻载时合并为2相工作(效率提升12%)
- 突发负载时拆分为独立8相(电流输出能力提升3倍)
-
零电流检测开关:采用栅极电荷回收技术,使开关损耗降低27%。具体实现是在每个MOSFET的驱动电路中加入电荷泵,将关断时的栅极电荷转移到即将开启的MOSFET上。
2.2 多域电源状态机
我们开发了名为"PowerMesh"的分布式状态管理架构,其核心是一个可编程的电源状态引擎(PSE)。这个32位RISC-V核专门负责处理电源状态转换,具有以下特点:
- 支持256种预定义电源模式
- 模式切换延迟<2μs
- 可并行处理最多8个电源域的独立状态转换
典型工作流程示例:
c复制// 切换到推理模式
pse_cmd(PS_DOMAIN_NPU, MODE_PERF); // NPU全速运行
pse_cmd(PS_DOMAIN_DDR, MODE_LOW_LAT); // 内存低延迟模式
pse_cmd(PS_DOMAIN_IO, MODE_SAFE); // IO保持安全电压
3. 实际调试中的挑战与解决方案
3.1 电源完整性优化
在首次硅验证时,我们遇到了严重的电源噪声问题:当NPU全速运行同时进行DDR5内存访问时,核心电压会出现周期性塌陷。通过以下措施最终解决:
-
PCB层叠优化:
- 将电源平面与地平面间距从6mil缩小到3mil
- 采用8层板设计,专为高速信号分配完整参考平面
-
去耦电容布局:
- 每平方厘米放置4个0201封装的100nF陶瓷电容
- 在BGA封装底部植球区域集成30个10μF硅电容
-
动态阻抗补偿:
python复制def impedance_compensation(vdroop): if vdroop > 50mV: increase_phase_count() elif vdroop < 10mV: decrease_phase_count()
3.2 热管理协同设计
PMU与散热系统的联动是另一个关键点。我们实现了温度-功率闭环控制:
- 在芯片内部布置16个温度传感器
- PMU每10ms读取一次温度分布图
- 动态调整电压频率曲线:
- 当局部温度>85℃时,自动降低该区域电压5%
- 整体温度>95℃时,触发全局降频协议
实测数据显示,这种方案使芯片在持续满载时:
- 最高温度降低11℃
- 性能损失仅3%
4. 量产测试中的特殊考量
4.1 自动化测试流程
我们开发了专门的PMU测试程序,包含37项自动化测试:
| 测试项目 | 标准值 | 容差范围 |
|---|---|---|
| 静态功耗 | 15mW | ±2mW |
| 瞬态响应 | <200ns | +50ns |
| 交叉调整率 | <3% | +1% |
测试中发现的一个典型问题:部分芯片在低温(-40℃)下启动时,LDO会进入振荡状态。最终发现是基准电压源的启动时序问题,通过修改POR(上电复位)电路解决。
4.2 可靠性强化措施
针对AI芯片7×24小时运行的特点,PMU加入了以下可靠性设计:
-
老化补偿机制:
- 实时记录每个电源模块的工作小时数
- 每1000小时自动提高输出电压0.5%以补偿MOSFET老化
-
故障预测系统:
- 监测关键参数(开关损耗、效率下降率)
- 当参数偏离基线>15%时触发预警
-
安全冗余设计:
- 所有关键控制信号都有双路径备份
- 采用三模冗余(TMR)的电压监测ADC
5. 未来演进方向
从AI-29项目中获得的最重要认知是:PMU正在从辅助单元变为决定AI芯片能效比的核心。我们正在研究几个突破性方向:
- 基于ML的负载预测:在PMU中集成微型神经网络,提前100μs预测负载变化
- 光子电源传输:实验性采用硅光技术进行芯片内部分布式供电
- 3D堆叠PMU:将电源管理电路垂直集成在计算单元下方,缩短供电距离
在最近的原型测试中,3D堆叠架构使供电网络阻抗降低了60%,这可能会彻底改变下一代AI芯片的电源设计范式。不过要真正量产,还需要解决散热和应力匹配等一系列工程挑战。
