NVIDIA开源GPU内核模块中follow_pfn函数移除的技术分析

背景介绍

在Linux内核6.9版本及后续6.10合并窗口中，内核开发团队移除了follow_pfn()函数，这一变更影响了NVIDIA开源GPU内核模块的构建过程。该函数原本用于通过虚拟内存区域(VMA)和地址获取物理页帧号(PFN)，是内存管理子系统的重要组成部分。

技术变更详情

Linux内核提交cb10c28ac82c9b7a5e9b3b1dc7157036c20c36dd移除了follow_pfn()函数，转而推荐使用更安全的follow_pte()函数组合。这一变更属于内核内存管理子系统的重构工作，目的是提供更精细的内存访问控制和更好的安全性。

对NVIDIA驱动的影响

NVIDIA开源GPU内核模块中的os-mlock.c文件使用了被移除的follow_pfn()函数，导致在6.9+内核上构建时出现编译错误。错误信息显示为隐式函数声明错误，因为内核头文件中已不再包含该函数的声明。

解决方案

NVIDIA开发团队提供了临时解决方案，建议用户修改os-mlock.c文件中的nv_follow_pfn()函数实现。修改后的版本使用follow_pte()函数组合来替代原有的follow_pfn()调用：

1. 首先检查虚拟内存区域的标志位，确认是否为IO或PFN映射区域
2. 使用follow_pte()获取页表项(PTE)和自旋锁
3. 通过pte_pfn()从PTE中提取物理页帧号
4. 最后释放页表映射和锁

需要注意的是，follow_pte()的第一个参数应为vma->vm_mm(内存管理结构)，而不是直接的VMA结构。

兼容性考虑

这一变更主要影响开发中的内核版本(如6.10合并窗口)，稳定版本内核用户通常不会遇到此问题。NVIDIA官方建议用户在稳定内核版本上使用其驱动，对于开发内核版本，用户可以自行应用补丁或等待官方更新。

技术原理深入

follow_pfn()函数的移除反映了Linux内核内存管理子系统向更精细控制方向的演进。新的follow_pte()方法提供了对页表项的直接访问，使开发者能够实现更复杂的内存管理逻辑，同时也增强了安全性，因为它明确要求处理页表锁。

这种变更也体现了内核开发中"显式优于隐式"的设计哲学，要求驱动开发者更清楚地表达他们的内存访问意图，而不是依赖旧函数提供的抽象层。

总结

内核API的变更是Linux开发的常态，NVIDIA开源GPU内核模块需要不断适应这些变化。对于开发者而言，理解这些变更背后的设计理念和技术细节，有助于更好地维护和开发内核模块。目前，用户可以通过手动修改代码或等待官方更新来解决这一兼容性问题。