Skip to content
JAX Array API

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

Sign in to view logs

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 #8672

Triggered via pull request January 18, 2025 01:16
@copybara-servicecopybara-service[bot]
synchronize #25974
test_716841435
Status Success
Total duration 3m 6s
Artifacts –

jax-array-api.yml

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

Annotations

1 warning
build (3.11)
ubuntu-latest pipelines will use ubuntu-24.04 soon. For more details, see https://github.com/actions/runner-images/issues/10636