jax
468a1e9a - [pallas:mosaic] Do not handle JAX memory spaces in the lowering

Commit

41 days ago

[pallas:mosaic] Do not handle JAX memory spaces in the lowering I don't think we can get an abstract value with such memory space ever? PiperOrigin-RevId: 885858961

References

#36017 - [pallas:mosaic] Do not handle JAX memory spaces in the lowering

Author

superbobry

Committer

Google-ML-Automation

Parents

ec97111d

jax 468a1e9a - [pallas:mosaic] Do not handle JAX memory spaces in the lowering

jax
468a1e9a - [pallas:mosaic] Do not handle JAX memory spaces in the lowering