xla

c7ef4395 - Subset of existing, approved PR (#9204)

Subset of existing, approved PR (#9204) approval from Zhanyong to merge despite GPU cicd test failure due to non-required, flaky status of GPU tests.