Skip to content
Back to pull request #6785

[Pallas] Make a FlashAttention Wrapper #6454

Sign in to view logs

[Pallas] Make a FlashAttention Wrapper

[Pallas] Make a FlashAttention Wrapper #6454

Run time

Learn about OS pricing on GitHub Actions
Job Run time
Build XLA / build
54m 46s
Build & publish docs
0s
CPU tests / test (benchmark_tests)
10m 42s
CPU tests / test (cpp_tests1)
15m 19s
CPU tests / test (cpp_tests2)
17m 22s
GPU tests / test (benchmark_tests)
11m 39s
CPU tests / test (python_tests, xla_op1)
20m 8s
GPU tests / test (cpp_tests1)
16m 17s
CPU tests / test (python_tests, xla_op2)
10m 37s
GPU tests / test (cpp_tests2)
25m 2s
CPU tests / test (python_tests, xla_op3)
12m 41s
GPU tests / test (python_tests, xla_op1)
1h 15m 13s
CPU tests / test (python_tests, torch_mp_op)
25m 50s
GPU tests / test (python_tests, xla_op2)
47m 19s
GPU tests / test (python_tests, xla_op3)
53m 19s
GPU tests / test (python_tests, torch_mp_op)
2h 6m 5s
8h 42m 19s