adjust benchmark size and add roofline info

kendryte · Nov 28, 2024 · 4f847af · 4f847af
1 parent 6d7a89d
commit 4f847af
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 20 deletions.
diff --git a/ntt/test/benchmark_test/benchmark_ntt.py b/ntt/test/benchmark_test/benchmark_ntt.py
@@ -287,16 +287,16 @@ def __init__(self, target: str, bin_path: str):
                                          'CH': '2.5',
                                          'HW': '2.0',
                                          },
-                              'expand': {'N': '0.0',
-                                         'C': '0.0',
-                                         'H': '0.0',
-                                         'W': '0.0',
-                                         'NC': '0.0',
-                                         'CH': '0.0',
-                                         'HW': '0.0',
-                                         'NH': '0.0',
-                                         'CW': '0.0',
-                                         'NW': '0.0',
+                              'expand': {'N': '1.5',
+                                         'C': '1.5',
+                                         'H': '1.5',
+                                         'W': '1.5',
+                                         'NC': '1.5',
+                                         'CH': '1.5',
+                                         'HW': '1.5',
+                                         'NH': '1.5',
+                                         'CW': '1.5',
+                                         'NW': '1.5',
                                          },
                               }
 

diff --git a/ntt/test/benchmark_test/benchmark_ntt_expand.cpp b/ntt/test/benchmark_test/benchmark_ntt_expand.cpp
@@ -70,14 +70,14 @@ int main(int argc, char *argv[]) {
     (void)argc;
     (void)argv;
 
-    benchmark_ntt_expand<1, 16, 16, 16, 16, 16, 16, 16>("N");
-    benchmark_ntt_expand<16, 1, 16, 16, 16, 16, 16, 16>("C");
-    benchmark_ntt_expand<16, 16, 1, 16, 16, 16, 16, 16>("H");
-    benchmark_ntt_expand<16, 16, 16, 1, 16, 16, 16, 16>("W");
-    benchmark_ntt_expand<1, 1, 16, 16, 16, 16, 16, 16>("NC");
-    benchmark_ntt_expand<16, 1, 1, 16, 16, 16, 16, 16>("CH");
-    benchmark_ntt_expand<16, 16, 1, 1, 16, 16, 16, 16>("HW");
-    benchmark_ntt_expand<1, 16, 1, 16, 16, 16, 16, 16>("NH");
-    benchmark_ntt_expand<16, 1, 16, 1, 16, 16, 16, 16>("CW");
-    benchmark_ntt_expand<1, 16, 16, 1, 16, 16, 16, 16>("NW");
+    benchmark_ntt_expand<1, 32, 32, 32, 32, 32, 32, 32>("N");
+    benchmark_ntt_expand<32, 1, 32, 32, 32, 32, 32, 32>("C");
+    benchmark_ntt_expand<32, 32, 1, 32, 32, 32, 32, 32>("H");
+    benchmark_ntt_expand<32, 32, 32, 1, 32, 32, 32, 32>("W");
+    benchmark_ntt_expand<1, 1, 32, 32, 32, 32, 32, 32>("NC");
+    benchmark_ntt_expand<32, 1, 1, 32, 32, 32, 32, 32>("CH");
+    benchmark_ntt_expand<32, 32, 1, 1, 32, 32, 32, 32>("HW");
+    benchmark_ntt_expand<1, 32, 1, 32, 32, 32, 32, 32>("NH");
+    benchmark_ntt_expand<32, 1, 32, 1, 32, 32, 32, 32>("CW");
+    benchmark_ntt_expand<1, 32, 32, 1, 32, 32, 32, 32>("NW");
 }