mm: restore KERNEL_DS optimisations