Skip to content
Lint

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

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

Run time

Learn about OS pricing on GitHub Actions
Job Run time
JAX C++
21s
JAX Python
22s
PaddlePaddle C++
19s
PaddlePaddle Python
58s
PyTorch C++
20s
PyTorch Python
1m 56s
4m 16s