ENH: multi gpu run