jax
56145d0a - [Mosaic TPU] Fix tpu.bitcast_vreg fold rule when input def op is null.

Commit

25 days ago

[Mosaic TPU] Fix tpu.bitcast_vreg fold rule when input def op is null. This can happen when input is block arg which does not have def op. PiperOrigin-RevId: 862849793

References

#34688 - [Mosaic TPU] Fix tpu.bitcast_vreg fold rule when input def op is null.

Author

bythew3i

Committer

Google-ML-Automation

Parents

f774eda1

jax 56145d0a - [Mosaic TPU] Fix tpu.bitcast_vreg fold rule when input def op is null.

jax
56145d0a - [Mosaic TPU] Fix tpu.bitcast_vreg fold rule when input def op is null.