Skip to content
Lint

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

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 #6542

Run time

Learn about OS pricing on GitHub Actions
Job Run time
PaddlePaddle C++
18s
JAX C++
19s
PaddlePaddle Python
52s
JAX Python
21s
PyTorch C++
23s
PyTorch Python
2m 1s
4m 14s