select_algorithm.py | 22 +- torch/nn/attention/_flex_attention.py | 265 ++++-- torch/testing/_internal/hop_db.py | 36 +- 9 files changed, 1159 insertions(+), 564 deletions(-) diff --git a/test/inductor/test_flex_attention.py b/test/inductor/test_flex_attention.py index abeee647...