jax
0fb278a0 - If sharding is not None (that's passed to convert_element_type), only compare it with operand's sharding if the sharding is concrete. Otherwise doing `getattr(operand, 'sharding')` on a `Tracer` leads to weird timeouts.

Commit

1 year ago

If sharding is not None (that's passed to convert_element_type), only compare it with operand's sharding if the sharding is concrete. Otherwise doing `getattr(operand, 'sharding')` on a `Tracer` leads to weird timeouts. PiperOrigin-RevId: 723595960

References

#227 - CI: 02/10/25 upstream sync

Author

yashk2810

Committer

Google-ML-Automation

Parents

f43d2b68

jax 0fb278a0 - If sharding is not None (that's passed to convert_element_type), only compare it with operand's sharding if the sharding is concrete. Otherwise doing `getattr(operand, 'sharding')` on a `Tracer` leads to weird timeouts.

jax
0fb278a0 - If sharding is not None (that's passed to convert_element_type), only compare it with operand's sharding if the sharding is concrete. Otherwise doing `getattr(operand, 'sharding')` on a `Tracer` leads to weird timeouts.