[flex-attention] Fluctuations in results for individual shapes

[Egor-Krivov]

On average chart is stable, but individual shapes have high variance. Maybe somewhere in a pytorch stack there is randomness in config choice?

Average chart:

Individual chart

[whitneywhtsang]

You can use TORCH_LOGS="+inductor" env var to print the selected config.