From 8805c02d68776bbed65e25cb1856a8f959f6ed26 Mon Sep 17 00:00:00 2001
From: "Dmitrichenko, Aleksei" <aleksei.dmitrichenko@intel.com>
Date: Tue, 14 Oct 2025 13:29:47 +0200
Subject: [PATCH] Add one autotuning config to the flex attention benchmark

---
 .../flex_attention_benchmark_causal_mask.py                      | 1 +
 1 file changed, 1 insertion(+)

diff --git a/benchmarks/triton_kernels_benchmark/flex_attention_benchmark_causal_mask.py b/benchmarks/triton_kernels_benchmark/flex_attention_benchmark_causal_mask.py
index 746dc57734..ed49066a2c 100644
--- a/benchmarks/triton_kernels_benchmark/flex_attention_benchmark_causal_mask.py
+++ b/benchmarks/triton_kernels_benchmark/flex_attention_benchmark_causal_mask.py
@@ -25,6 +25,7 @@
 def get_flex_attn_fwd_configs(*args, **kwargs):  # pylint: disable=unused-argument
     configs = [
         FlexConfig(32, 16, 2, 4),
+        FlexConfig(64, 32, 2, 4),
         FlexConfig(128, 64, 2, 16),
         FlexConfig(128, 64, 2, 8),
         FlexConfig(128, 32, 2, 16),