mm/highmem: prepare for overriding set_pte_at()