GPU显存管理：原理、优化与实践指南

1. 显存管理基础概念解析

在GPU内核模式驱动（KMD）开发中，显存管理是最核心的模块之一。与常规系统内存不同，显存（Video RAM）具有特殊的访问特性和性能特征。现代GPU通常配备GDDR6或HBM2等高带宽存储器，其物理特性决定了必须采用专门的管理策略。

显存分配器需要处理三个关键维度：

• 物理地址连续性要求（影响DMA传输效率）
• 内存对齐约束（通常需要64KB或2MB对齐）
• 内存类型选择（本地显存、系统内存或分页内存）

典型的显存分配流程会经过以下层级：

1. 应用程序通过API（如Vulkan或CUDA）请求内存
2. 用户模式驱动（UMD）转换为标准内存描述符
3. KMD根据硬件特性执行实际分配
4. 内存管理单元（MMU）建立页表映射

关键提示：现代GPU普遍采用统一内存架构（UMA），但"显存"这个术语仍被广泛使用，实际可能指代任何可供GPU访问的内存区域。

2. 显存分配器实现细节

2.1 物理内存管理

以Linux DRM子系统为例，核心数据结构如下：

c复制struct drm_gem_object {
    struct kref refcount;
    size_t size;
    struct address_space *filp;
    struct drm_device *dev;
    struct list_head vmas;
};

内存分配通常通过drm_gem_private_object_init()初始化对象，再调用平台特定的分配函数（如amdgpu_bo_create()）。关键参数包括：

• 请求大小（需按硬件页大小对齐）
• 内存域（VRAM/SYSTEM/CPU_GTT）
• 缓存策略（WC/UC/WT）
• 分配标志（可压缩、可加密等）

2.2 虚拟地址映射

GPU使用多级页表实现VA到PA的转换，现代架构通常采用4级或5级页表。映射建立过程涉及：

1. 分配GPU虚拟地址（通过drm_mm_alloc）
2. 创建页表项（PTE）
3. 配置MMU寄存器
4. 必要时刷新TLB

典型映射代码路径：

bash复制amdgpu_vm_bo_map()
└─ amdgpu_vm_update_range()