Skip to content
Build and test

[Pallas] Support segment ids in flash attention #8021

Sign in to view logs

[Pallas] Support segment ids in flash attention

[Pallas] Support segment ids in flash attention #8021

Run time

Learn about OS pricing on GitHub Actions
Job Run time
Build upstream CI image / build
56m 25s
Build XLA CUDA plugin / build
4m 36s
Build PyTorch/XLA / build
22m 44s
TPU tests
0s
CPU tests / test (benchmark_tests)
2m 9s
Build docs / build-docs
1m 40s
CPU tests / test (python_tests, xla_op1)
11m 8s
GPU tests / test (benchmark_tests)
5m 21s
CPU tests / test (python_tests, xla_op2)
2m 37s
GPU tests / test (python_tests, xla_op1)
27m 15s
CPU tests / test (python_tests, xla_op3)
3m 59s
GPU tests / test (python_tests, xla_op2)
5m 9s
CPU tests / test (python_tests, torch_mp_op)
15m 11s
GPU tests / test (python_tests, xla_op3)
40m 38s
CPU tests / test (cpp_tests, cpp_tests1)
3m 4s
GPU tests / test (python_tests, torch_mp_op)
47m 13s
CPU tests / test (cpp_tests, cpp_tests2)
3m 46s
GPU tests / test (cpp_tests, cpp_tests1)
5m 0s
GPU tests / test (cpp_tests, cpp_tests2)
15m 15s
4h 33m 10s