jax
3e8255fc - Bring back `os.getenv` for `jax_dump_ir_to`, `jax_include_debug_info_in_dumps` and `jax_dump_ir_modes` because they are flags and not states. `env-var` by default only works for states.

Commit

307 days ago

Bring back `os.getenv` for `jax_dump_ir_to`, `jax_include_debug_info_in_dumps` and `jax_dump_ir_modes` because they are flags and not states. `env-var` by default only works for states. PiperOrigin-RevId: 778275880

References

#29916 - Bring back `os.getenv` for `jax_dump_ir_to`, `jax_include_debug_info_in_dumps` and `jax_dump_ir_modes` because they are flags and not states. `env-var` by default only works for states.

Author

yashk2810

Committer

Google-ML-Automation

Parents

d241f6d6

jax 3e8255fc - Bring back `os.getenv` for `jax_dump_ir_to`, `jax_include_debug_info_in_dumps` and `jax_dump_ir_modes` because they are flags and not states. `env-var` by default only works for states.

jax
3e8255fc - Bring back `os.getenv` for `jax_dump_ir_to`, `jax_include_debug_info_in_dumps` and `jax_dump_ir_modes` because they are flags and not states. `env-var` by default only works for states.