jax
85af862e - [Try again] For nested pjit's cache the generation of StableHLO if it satifies the key. This should help in improving the lowering time.

Commit

2 years ago

[Try again] For nested pjit's cache the generation of StableHLO if it satifies the key. This should help in improving the lowering time. Reverts 4a5c6f82009dee9c30ca4a85359a702d745ed035 PiperOrigin-RevId: 577974380

Author

yashk2810

Committer

a-googler

Parents

7eddc76a

jax 85af862e - [Try again] For nested pjit's cache the generation of StableHLO if it satifies the key. This should help in improving the lowering time.

jax
85af862e - [Try again] For nested pjit's cache the generation of StableHLO if it satifies the key. This should help in improving the lowering time.