add ntt cases in default.nix

chipsalliance · Jan 22, 2025 · 6f971dd · 6f971dd
1 parent ec86041
commit 6f971dd
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/tests/eval/_ntt/default.nix b/tests/eval/_ntt/default.nix
@@ -33,6 +33,12 @@ in {
   ntt_256 = build_ntt "ntt_256" ./ntt.c ./ntt_256_main.c;
   ntt_512 = build_ntt "ntt_512" ./ntt.c ./ntt_512_main.c;
   ntt_1024 = build_ntt "ntt_1024" ./ntt.c ./ntt_1024_main.c;
+  ntt_4096 = build_ntt "ntt_4096" ./ntt.c ./ntt_4096_main.c;
+
+  ntt_mem_64 = build_ntt "ntt_mem_64" ./ntt_mem.c ./ntt_64_main.c;
+  ntt_mem_128 = build_ntt "ntt_mem_128" ./ntt_mem.c ./ntt_128_main.c;
+  ntt_mem_256 = build_ntt "ntt_mem_256" ./ntt_mem.c ./ntt_256_main.c;
+  ntt_mem_512 = build_ntt "ntt_mem_512" ./ntt_mem.c ./ntt_512_main.c;
   ntt_mem_1024 = build_ntt "ntt_mem_1024" ./ntt_mem.c ./ntt_1024_main.c;
   ntt_mem_4096 = build_ntt "ntt_mem_4096" ./ntt_mem.c ./ntt_4096_main.c;
 }
diff --git a/tests/eval/_ntt/ntt_512_main.c b/tests/eval/_ntt/ntt_512_main.c
@@ -60,8 +60,8 @@ void test() {
       761,   10180, 6570,  6349,  9028,  10495, 4756,  9332,  8348,  4995,
       6933,  4351,  111,   1610,  7410,  960,   11972, 2853,  3551,  1423,
       2634,  3972};
-  const int twiddle[9] = {3400, 8340, 12149, 7311,
-                           5860,  4134, 8246, 1479,  12288};
+  // const int twiddle[9] = {3400, 8340, 12149, 7311,
+                           // 5860,  4134, 8246, 1479,  12288};
   const int twiddle[] = {
     // layer #0
     1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,