gpu: mimic wrap option in schedule tree