Skip to content
CI

Clean up some code in pxla.py that deals with jaxpr and avals. Lift the discharging of refs into a separate function and remove global_in_avals argument from lower_sharding_computation #50755

Sign in to view logs

Clean up some code in pxla.py that deals with jaxpr and avals. Lift the discharging of refs into a separate function and remove global_in_avals argument from lower_sharding_computation

Clean up some code in pxla.py that deals with jaxpr and avals. Lift the discharging of refs into a separate function and remove global_in_avals argument from lower_sharding_computation #50755

Triggered via pull request April 27, 2024 00:54
@copybara-servicecopybara-service[bot]
synchronize #20963
test_628534749
Status Success
Total duration 10m 2s
Billable time 18m
Artifacts –

ci-build.yaml

on: pull_request
lint_and_typecheck
1m 23s
lint_and_typecheck
Matrix: build
Matrix: Documentation - test code snippets
Matrix: Documentation - render documentation
Fit to window
Zoom out
Zoom in