xla

aa0b6602 - Disable test_index_put_accumulate_large_tensor_xla on GPU due to memory issue (#2625)