PR #10185: Hook up host offloading passes on GPU
PR #10185: Hook up host offloading passes on GPU

Imported from GitHub PR https://github.com/openxla/xla/pull/10185

This adds the host memory offloading passes to the GPU compiler.

With this change, some of the Jax tests newly pass, see https://github.com/google/jax/pull/20036 for the relevant tests.
Copybara import of the project:

--
cc65a9bcf7e4db861d637bb5e5471f666cf516bf by Jaroslav Sevcik <jsevcik@nvidia.com>:

Hook up offloading passes on GPU

--
69217f15a4bcec48676b892f4ec35ec8059beaab by Jaroslav Sevcik <jsevcik@nvidia.com>:

Add missing build dep

--
7ffb35b970427eadffcce2ff4a74f40e7a0e458d by Jaroslav Sevcik <jsevcik@nvidia.com>:

Format

Merging this change closes #10185

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/10185 from jaro-sevcik:hook-up-host-offloading-gpu 7ffb35b970427eadffcce2ff4a74f40e7a0e458d
