Skip to content
CI - Cloud TPU (presubmit)

Put abstract_mesh on every eqn so that we can preserve it during eval_jaxpr and check_jaxpr roundtrip. #102

Sign in to view logs

Put abstract_mesh on every eqn so that we can preserve it during eval_jaxpr and check_jaxpr roundtrip.

Put abstract_mesh on every eqn so that we can preserve it during eval_jaxpr and check_jaxpr roundtrip. #102

Run time

Learn about OS pricing on GitHub Actions
Job Run time
TPU test (jaxlib=head, v5e-8)
6m 3s
6m 3s