Logs a fatal message if Auto Sharding times out (since we no longer rely on GSPMD as a backup).
Logs a fatal message if Auto Sharding times out (since we no longer rely on GSPMD as a backup).

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/10759 from Intel-tensorflow:akhil/gemv 4534fbd7e96a6e45ccc5a501b38b940aa8bd9d38
