s390/mm: implement set_memory_4k()