update

intel · May 10, 2024 · a3dd01b · a3dd01b
1 parent 07b2d64
commit a3dd01b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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',