jax
d5bc1dd9 - If you trace and lower with abstract mesh and abstract inputs and then do `out = lowered.compile(device_assignment=...)(real_arr)`, then make sure the `out.sharding` is concrete i.e. has real devices and is not abstract.

Commit

16 days ago

If you trace and lower with abstract mesh and abstract inputs and then do `out = lowered.compile(device_assignment=...)(real_arr)`, then make sure the `out.sharding` is concrete i.e. has real devices and is not abstract. PiperOrigin-RevId: 938074332

Author

yashk2810

Committer

Google-ML-Automation

Parents

184e7bf4

jax d5bc1dd9 - If you trace and lower with abstract mesh and abstract inputs and then do `out = lowered.compile(device_assignment=...)(real_arr)`, then make sure the `out.sharding` is concrete i.e. has real devices and is not abstract.

jax
d5bc1dd9 - If you trace and lower with abstract mesh and abstract inputs and then do `out = lowered.compile(device_assignment=...)(real_arr)`, then make sure the `out.sharding` is concrete i.e. has real devices and is not abstract.