add roofline info for pack hw

kendryte · Nov 18, 2024 · 4bf1fe5 · 4bf1fe5
1 parent ef19cb3
commit 4bf1fe5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ntt/test/benchmark_test/benchmark_ntt.py b/ntt/test/benchmark_test/benchmark_ntt.py
@@ -277,7 +277,7 @@ def __init__(self, target: str, bin_path: str):
                                        'W': '1.5',
                                        'NC': '2.5',
                                        'CH': '2.5',
-                                       'HW': '1.5',
+                                       'HW': '2.0',
                                        },
                               'unpack': {'N': '0',
                                          'C': '0',

diff --git a/ntt/test/benchmark_test/benchmark_ntt_pack.cpp b/ntt/test/benchmark_test/benchmark_ntt_pack.cpp
@@ -101,7 +101,7 @@ int main(int argc, char *argv[]) {
                                                                            1);
     benchmark_ntt_pack<ntt::vector<float, P, P>, 2, 4 * P, 8 * P, 8, 1, 2>(
         "CH", 2000);
-    benchmark_ntt_pack<ntt::vector<float, P, P>, 2, 2, 8 * P, 8 * P, 2, 3>(
+    benchmark_ntt_pack<ntt::vector<float, P, P>, 4, 4, 8 * P, 8 * P, 2, 3>(
         "HW", 2000);
 #else
     benchmark_ntt_pack<ntt::vector<float, P>, 16 * P, 3, 4, 4, 0>("N", 300);