[S390] Change vmalloc defintions