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. #1222

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. #1222

Triggered via pull request January 7, 2025 19:03
@copybara-servicecopybara-service[bot]
synchronize #25742
test_712664646
Status Success
Total duration 6m 26s
Artifacts –

bazel_gpu_rbe.yml

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