Support xN-transpose benchmarking for problem sizes HW128, 256, and 512.

Add support for different problem sizes in the xN transpose benchmark. Presently, the benchmarking covers problem sizes of 1xx, with the subsequent step being 1024. This gap hinders the comprehensive evaluation of the correlation between tile size and problem size. This commit include problem sizes 128, 256, and 512 to provide a more nuanced understanding of the relationship between tile size and diverse problem sizes. Signed-off-by: Phoebe Chen <[email protected]>
phoebesv · Dec 14, 2023 · 5101297 · 5101297
1 parent 50b08e4
commit 5101297
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 0 deletions.
diff --git a/bench/x16-transpose.cc b/bench/x16-transpose.cc
@@ -62,6 +62,9 @@ static void BenchmarkKernelSize(benchmark::internal::Benchmark* b)
   b->Args({32, 32});
   b->Args({64, 64});
   b->Args({117, 117});
+  b->Args({128, 128});
+  b->Args({256, 256});
+  b->Args({512, 512});
   b->Args({1024, 1024});
 }
 

diff --git a/bench/x32-transpose.cc b/bench/x32-transpose.cc
@@ -62,6 +62,8 @@ static void BenchmarkKernelSize(benchmark::internal::Benchmark* b)
   b->Args({32, 32});
   b->Args({64, 64});
   b->Args({128, 128});
+  b->Args({256, 256});
+  b->Args({512, 512});
   b->Args({1024, 1024});
 }
 

diff --git a/bench/x64-transpose.cc b/bench/x64-transpose.cc
@@ -62,6 +62,9 @@ static void BenchmarkKernelSize(benchmark::internal::Benchmark* b)
   b->Args({32, 32});
   b->Args({64, 64});
   b->Args({117, 117});
+  b->Args({128, 128});
+  b->Args({256, 256});
+  b->Args({512, 512});
   b->Args({1024, 1024});
 }
 

diff --git a/bench/x8-transpose.cc b/bench/x8-transpose.cc
@@ -62,6 +62,9 @@ static void BenchmarkKernelSize(benchmark::internal::Benchmark* b)
   b->Args({32, 32});
   b->Args({64, 64});
   b->Args({117, 117});
+  b->Args({128, 128});
+  b->Args({256, 256});
+  b->Args({512, 512});
   b->Args({1024, 1024});
 }