Skip to content

Fix a bug in flash attention where kv_seq_len should divide block_k_major. #10981

Fix a bug in flash attention where kv_seq_len should divide block_k_major.

Fix a bug in flash attention where kv_seq_len should divide block_k_major. #10981

CPU tests  /  test (python_tests, xla_op1)

succeeded Feb 7, 2025 in 18m 22s