pmdp_invalidate_ad是 Linux 内核中用于在原子上下文中使一个 PMD 条目失效的函数专门服务于透明大页THP的自动NUMA平衡AutoNUMA场景。核心作用使 PMD 失效而不刷新 TLB这个函数的核心操作是pmdp_establish(vma, address, pmdp, pmd_mkinvalid(*pmdp))它将 PMD 条目标记为“无效”但不执行 TLB 刷新。关键在于注释中的解释“一旦建立了无效的 PTE其访问位和脏位就无法再被更新。”这意味着CPU 在 TLB 中可能还缓存着旧的 PMD 条目但由于新的 PMD 是无效的任何后续的访问都会触发缺页异常而不是通过旧的 TLB 条目静默地更新访问/脏位。因此不需要立即刷新 TLB可以推迟到更合适的时机。参数与调用场景pmd_t pmdp_invalidate_ad(struct vm_area_struct *vma, unsigned long address, pmd_t *pmdp)vma对应的虚拟内存区域。address要失效的虚拟地址PMD 对齐。pmdp指向 PMD 条目的指针。这个函数主要在do_huge_pmd_numa_page()中被调用。当 AutoNUMA 决定将一个 THP 页面从当前 NUMA 节点迁移到另一个节点时需要先使旧的 PMD 映射失效防止在迁移过程中有其他 CPU 通过旧映射访问该页面。与pmdp_invalidate的区别内核中还有一个更通用的pmdp_invalidate()函数。两者的核心区别在于是否刷新 TLB函数TLB 刷新适用场景pmdp_invalidate_ad()不刷新AutoNUMA 迁移依赖“无效 PTE 阻止脏位更新”的硬件特性pmdp_invalidate()刷新通用场景需要立即确保 TLB 一致性_ad后缀正是“AutoNUMA”的缩写表明这个函数是为 AutoNUMA 的特定需求而设计的。设计意图这个函数体现了内核对硬件行为的精细利用。在 AutoNUMA 迁移中如果每次使 PMD 失效都立即刷新 TLB开销会很大。而利用“无效 PTE 无法更新脏位”这一特性内核可以安全地推迟 TLB 刷新将多个失效操作合并处理从而优化性能。调用关系pmdp_invalidate_ad的调用关系很集中它主要是为change_huge_pmd()这个函数服务的。唯一调用者pmdp_invalidate_ad()在代码中只被change_huge_pmd()调用。它的引入本身就是为了替换change_huge_pmd()中原有的pmdp_invalidate()调用以消除不必要的 TLB 刷新。背景与设计意图这个替换源于一个性能优化在 THP透明大页上调用change_protection_range()时x86 架构会触发两次 TLB 刷新——一次在change_huge_pmd()内部立即刷新另一次在change_protection_range()结束时批量刷新。第一次刷新原本是为了防止硬件在 PMD 被修改期间更新脏位/访问位。但 x86 硬件在设置脏位时会原子性地检查 PTE 是否仍然存在因此这个刷新是不必要的Intel Knights Landing 除外。pmdp_invalidate_ad()正是利用这一硬件特性只使 PMD 失效而不刷新 TLB从而让后续的批量刷新可以合并操作。间接依赖pmdp_invalidate_ad()内部调用了pmdp_establish()。在 x86 上pmdp_establish()使用cmpxchg64原子地替换 PMD 条目并返回旧值。如果检测到 CPU 存在X86_BUG_PTE_LEAK即 Knights Landing 的硬件缺陷则会回退到执行一次 TLB 刷新。pmdp_invalidate_ad是change_huge_pmd的专属优化通过利用 x86 硬件的原子性保证省去了 THP 保护变更路径上一次多余的 TLB 刷新。