xla

1fe2a949 - Revert "Adopt new changes in upstream"

Revert "Adopt new changes in upstream" as upstream reverts their change in pytorch/pytorch#99168. This reverts commit 4412fc2278d8c8008f40f8efd37466ad5a4543ba.