[cases] eval.ntt* fix correctness, add 64 and 4096 config

chipsalliance · Jan 6, 2025 · 9d2a214 · 9d2a214
1 parent b445366
commit 9d2a214
Show file tree

Hide file tree

Showing 5 changed files with 473 additions and 7 deletions.
diff --git a/tests/eval/_ntt/default.nix b/tests/eval/_ntt/default.nix
@@ -28,9 +28,11 @@ let
     };
 
 in {
+  ntt_64 = build_ntt "ntt_64" ./ntt.c ./ntt_64_main.c;
   ntt_128 = build_ntt "ntt_128" ./ntt.c ./ntt_128_main.c;
   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_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/gen_data.py b/tests/eval/_ntt/gen_data.py
@@ -2,9 +2,9 @@
 
 def main():
     vlen = 4096
-    l = 10
+    l = 12
     n = 1 << l
-    assert n <= vlen // 4
+    # assert n <= vlen // 4
     p = 12289  # p is prime and n | p - 1
     g = 11  # primitive root of p
     assert (p - 1) % n == 0