From 6f971dd340d74e0387c69e2a08c326307fee31b4 Mon Sep 17 00:00:00 2001
From: Lucas-Wye <lustrew@foxmail.com>
Date: Wed, 22 Jan 2025 09:57:27 +0000
Subject: [PATCH] add ntt cases in default.nix

---
 tests/eval/_ntt/default.nix    | 6 ++++++
 tests/eval/_ntt/ntt_512_main.c | 4 ++--
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/tests/eval/_ntt/default.nix b/tests/eval/_ntt/default.nix
index 18fd15c33..77a1a7a0b 100644
--- 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
index 270ed3636..b12689bfc 100644
--- 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,