ion: Port heap mask change to ion

Heap mask field passes as argument at allocation time to
specify ion heap.