Reverts f6a2b933193c44bf382e47e099122deccc4d2f8c
Reverts f6a2b933193c44bf382e47e099122deccc4d2f8c

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/12465 from openxla:fix_flag 4a050dce154b924b98134294f3f15ffce35bdcbb
