jax
3a54f93d - [Pallas][Mosaic][TPU] Add `disable_bounds_checks` compiler params

Commit

1 year ago

[Pallas][Mosaic][TPU] Add `disable_bounds_checks` compiler params When we run the program with "--xla_jf_bounds_check=true", we can selectively disable bounds checks for pallas kernels now. PiperOrigin-RevId: 748193719

References

#392 - CI: 04/30/25 upstream sync

#518 - [0.6.0-UT] Adjust tolerance for reciprocal in unary primitives test

#519 - [0.6.0-UT] Fix Test Shard

#594 - Add CI workflow for Pytest ROCm

Author

bythew3i

Committer

charleshofer

Parents

3ad33334

jax 3a54f93d - [Pallas][Mosaic][TPU] Add `disable_bounds_checks` compiler params

jax
3a54f93d - [Pallas][Mosaic][TPU] Add `disable_bounds_checks` compiler params