Skip to content
Lint

[JAX] Correct fused attention output after each step of ring attention #6546

Sign in to view logs

[JAX] Correct fused attention output after each step of ring attention

[JAX] Correct fused attention output after each step of ring attention #6546

Triggered via pull request January 8, 2025 16:08
@mgoldfarb-nvidiamgoldfarb-nvidia
synchronize #1393
mgoldfarb-nvidia:mgoldfarb-nvidia/online_softmax_aux_correction
Status Success
Total duration 2m 4s
Artifacts –

lint.yml

on: pull_request
PyTorch C++
24s
PyTorch C++
PyTorch Python
1m 55s
PyTorch Python
JAX C++
25s
JAX C++
JAX Python
24s
JAX Python
PaddlePaddle C++
21s
PaddlePaddle C++
PaddlePaddle Python
1m 2s
PaddlePaddle Python
Fit to window
Zoom out
Zoom in

Annotations

6 warnings
PaddlePaddle C++
ubuntu-latest pipelines will use ubuntu-24.04 soon. For more details, see https://github.com/actions/runner-images/issues/10636
JAX C++
ubuntu-latest pipelines will use ubuntu-24.04 soon. For more details, see https://github.com/actions/runner-images/issues/10636
PyTorch C++
ubuntu-latest pipelines will use ubuntu-24.04 soon. For more details, see https://github.com/actions/runner-images/issues/10636
JAX Python
ubuntu-latest pipelines will use ubuntu-24.04 soon. For more details, see https://github.com/actions/runner-images/issues/10636
PaddlePaddle Python
ubuntu-latest pipelines will use ubuntu-24.04 soon. For more details, see https://github.com/actions/runner-images/issues/10636
PyTorch Python
ubuntu-latest pipelines will use ubuntu-24.04 soon. For more details, see https://github.com/actions/runner-images/issues/10636