Modernize tensor contraction code: bug fixes, dead code removal, and cleanup

libeigen/eigen!2248 Co-authored-by: Rasmus Munk Larsen <rmlarsen@gmail.com>
2026-04-10 11:34:33 +08:00 · 2026-03-29 18:03:06 -07:00
parent 732ebc8cc2
commit 09581fda38
5 changed files with 133 additions and 156 deletions
--- a/unsupported/benchmarks/Tensor/bench_contraction.cpp
+++ b/unsupported/benchmarks/Tensor/bench_contraction.cpp
@@ -131,7 +131,7 @@ static void ContractionSizes(::benchmark::Benchmark* b) {

 static void ThreadPoolSizes(::benchmark::Benchmark* b) {
  for (int size : {64, 256, 512, 1024}) {
-    for (int threads : {2, 4, 8}) {
+    for (int threads : {1, 2, 4, 8, 16}) {
      b->Args({size, size, size, threads});
    }
  }