xla

844856d2 - Backport #7329: Avoid OOM when build PyTorch with CUDA enabled (#7618)