raiseException('Number of pixels in the image is not divisible by the number of GPUs!\n\t# pixels: {}\n\t# GPUs: {}'.format(ray_batch['ray_d'].shape[0],
world_size))
# split into ranks; make sure different processes don't overlap
# split into ranks; make sure different processes don't overlap