xla

c91b7661 - Additional fixes for xmp and gpu build (#3410)

Additional fixes for xmp and gpu build (#3410) * Update CUDA capability to cover K80 * Fix xla_multiprocessing._setup_workers (GPU) parsing error