Skip to content

Add a sharding rule for reduce_precision_p and properly thread eqn.ctx in loops.py where we create pe.new_jaxpr_eqn's #2018

Add a sharding rule for reduce_precision_p and properly thread eqn.ctx in loops.py where we create pe.new_jaxpr_eqn's

Add a sharding rule for reduce_precision_p and properly thread eqn.ctx in loops.py where we create pe.new_jaxpr_eqn's #2018

Triggered via pull request January 18, 2025 01:31
Status Success
Total duration 4m 0s
Artifacts

bazel_cpu_rbe.yml

on: pull_request
Matrix: run_tests
Fit to window
Zoom out
Zoom in