xla

cd09e19f - Revert "Companion PR for https://github.com/pytorch/pytorch/pull/83628 (#3914)" (#3932)