Skip to content
CI - Bazel GPU tests (RBE)

Sort manual axes when lowering jax.shard_map to sdy.manual_computation, which ensures the determinism in the generated sdy.manual_computation. #1220

Sign in to view logs

Sort manual axes when lowering jax.shard_map to sdy.manual_computation, which ensures the determinism in the generated sdy.manual_computation.

Sort manual axes when lowering jax.shard_map to sdy.manual_computation, which ensures the determinism in the generated sdy.manual_computation. #1220

Run time

Learn about OS pricing on GitHub Actions
Job Run time
Bazel single accelerator GPU tests (linux-x86-n2-16, Python 3.12, x64=1)
14m 1s
Bazel single accelerator GPU tests (linux-x86-n2-16, Python 3.12, x64=0)
13m 56s
27m 57s