Rollback https://github.com/openxla/xla/commit/0ab2be0b5a575da3206d2c2f92b85e6346708405.
Rollback https://github.com/openxla/xla/commit/0ab2be0b5a575da3206d2c2f92b85e6346708405.

There is an internal issue with running tests on H100s requiring the change to be rolled back.

Reverts 42883cb09d1a8155824ce4ed044794c0dffdd19f
