diff --git a/benchmarks/xetla_benchmark/fused_softmax.py b/benchmarks/xetla_benchmark/fused_softmax.py
index 8f1055ad43..5b33569fd2 100644
--- a/benchmarks/xetla_benchmark/fused_softmax.py
+++ b/benchmarks/xetla_benchmark/fused_softmax.py
@@ -118,8 +118,8 @@ def softmax(x):
 @triton.testing.perf_report(
     triton.testing.Benchmark(
         x_names=['N'],  # argument names to use as an x-axis for the plot
-        #x_vals=[256, 1024, 2048, 4096],  # different possible values for `x_name`
-        x_vals=[4096],  # different possible values for `x_name`        
+        x_vals=[256, 1024, 2048, 4096],  # different possible values for `x_name`
+        # x_vals=[4096],  # different possible values for `x_name`        
         line_arg='provider',  # argument name whose value corresponds to a different line in the plot
         line_vals=[
             'triton',