First attempt to use more gpus

A collaborative PR for making UFO work on more GPUs from concert.