Add runtime cache size detection for ARM and improve GEMM blocking

libeigen/eigen!2282

Co-authored-by: Rasmus Munk Larsen <rmlarsen@gmail.com>
This commit is contained in:
Rasmus Munk Larsen
2026-03-11 22:36:33 -07:00
parent 42c1dbd2c3
commit 8368a12f0f
4 changed files with 44 additions and 3 deletions

View File

@@ -35,7 +35,7 @@ static void BM_EigenGemm(benchmark::State& state) {
}
static void GemmSizes(::benchmark::Benchmark* b) {
for (int size : {8, 16, 32, 64, 96, 128, 160, 192, 224, 256, 288, 320, 384, 448, 512, 768, 1024, 1536, 2048}) {
for (int size : {8, 16, 32, 64, 96, 128, 160, 192, 224, 256, 288, 320, 384, 448, 512, 768, 1024, 1536, 2048, 4096}) {
b->Args({size, size, size});
}
// Non-square sizes