[eval] use #ifdef instead of commenting code

chipsalliance · Jan 26, 2025 · f51b47e · f51b47e
1 parent 804da11
commit f51b47e
Show file tree

Hide file tree

Showing 7 changed files with 54 additions and 48 deletions.
diff --git a/tests/eval/_ntt/ntt.c b/tests/eval/_ntt/ntt.c
@@ -1,9 +1,6 @@
 #include <assert.h>
 #include <stdio.h>
 
-// #define USERN 32
-// #define DEBUG
-
 // array is of length n=2^l, p is a prime number
 // roots is of length l, where g = roots[0] satisfies that
 //   g^(2^l) == 1 mod p  and g^(2^(l-1)) == -1 mod p
@@ -94,15 +91,15 @@ void ntt(const int *array, int l, const int *twiddle, int p, int *dst) {
         "vle32.v v12, 0(%5)\n"
 
     #ifdef DEBUG
-        "vse32.v v8, 0(%6)\n"// c
-        "vse32.v v12, 0(%7)\n"// c
-        "vse32.v v16, 0(%8)\n"// c
+        "vse32.v v8, 0(%6)\n"
+        "vse32.v v12, 0(%7)\n"
+        "vse32.v v16, 0(%8)\n"
     #endif
 
         // butterfly operation
         "vmul.vv v12, v12, v16\n"
         "vrem.vx v12, v12, %3\n"
-        "vadd.vv v16, v8, v12\n" // TODO: will it overflow?
+        "vadd.vv v16, v8, v12\n" // NOTE: use lazy reduction here
         "vsub.vv v20, v8, v12\n"
         // save half coefficients
         "vse32.v v16, 0(%4)\n"

diff --git a/tests/eval/_ntt/ntt_1024_main.c b/tests/eval/_ntt/ntt_1024_main.c
@@ -148,12 +148,14 @@ void test() {
   int dst[1024];
   ntt(arr, l, twiddle, p, dst);
 
-  // for (int i = 0; i < n; i++) {
-  //   printf("%d", dst[i]);
-  //   if ((i + 1) % 8 == 0) {
-  //     printf("\n");
-  //   } else {
-  //     printf(" ");
-  //   }
-  // }
+#ifdef DEBUG
+  for (int i = 0; i < n; i++) {
+    printf("%d", dst[i]);
+    if ((i + 1) % 8 == 0) {
+      printf("\n");
+    } else {
+      printf(" ");
+    }
+  }
+#endif
 }
diff --git a/tests/eval/_ntt/ntt_128_main.c b/tests/eval/_ntt/ntt_128_main.c
@@ -48,12 +48,14 @@ void test() {
   int dst[128];
   ntt(arr, l, twiddle, p, dst);
 
-  // for (int i = 0; i < n; i++) {
-  //   printf("%d", dst[i]);
-  //   if ((i + 1) % 8 == 0) {
-  //     printf("\n");
-  //   } else {
-  //     printf(" ");
-  //   }
-  // }
+#ifdef DEBUG
+  for (int i = 0; i < n; i++) {
+    printf("%d", dst[i]);
+    if ((i + 1) % 8 == 0) {
+      printf("\n");
+    } else {
+      printf(" ");
+    }
+  }
+#endif
 }
diff --git a/tests/eval/_ntt/ntt_256_main.c b/tests/eval/_ntt/ntt_256_main.c
@@ -64,12 +64,14 @@ void test() {
   int dst[256];
   ntt(arr, l, twiddle, p, dst);
 
-  // for (int i = 0; i < n; i++) {
-  //   printf("%d", dst[i]);
-  //   if ((i + 1) % 8 == 0) {
-  //     printf("\n");
-  //   } else {
-  //     printf(" ");
-  //   }
-  // }
+#ifdef DEBUG
+  for (int i = 0; i < n; i++) {
+    printf("%d", dst[i]);
+    if ((i + 1) % 8 == 0) {
+      printf("\n");
+    } else {
+      printf(" ");
+    }
+  }
+#endif
 }
diff --git a/tests/eval/_ntt/ntt_4096_main.c b/tests/eval/_ntt/ntt_4096_main.c
@@ -459,12 +459,14 @@ void test() {
   int dst[4096];
   ntt(arr, l, twiddle, p, dst);
 
-  // for (int i = 0; i < n; i++) {
-  //   printf("%d", dst[i]);
-  //   if ((i + 1) % 8 == 0) {
-  //     printf("\n");
-  //   } else {
-  //     printf(" ");
-  //   }
-  // }
+#ifdef DEBUG
+  for (int i = 0; i < n; i++) {
+    printf("%d", dst[i]);
+    if ((i + 1) % 8 == 0) {
+      printf("\n");
+    } else {
+      printf(" ");
+    }
+  }
+#endif
 }
diff --git a/tests/eval/_ntt/ntt_512_main.c b/tests/eval/_ntt/ntt_512_main.c
@@ -94,12 +94,14 @@ void test() {
   int dst[512];
   ntt(arr, l, twiddle, p, dst);
 
-  // for (int i = 0; i < n; i++) {
-  //   printf("%d", dst[i]);
-  //   if ((i + 1) % 8 == 0) {
-  //     printf("\n");
-  //   } else {
-  //     printf(" ");
-  //   }
-  // }
+#ifdef DEBUG
+  for (int i = 0; i < n; i++) {
+    printf("%d", dst[i]);
+    if ((i + 1) % 8 == 0) {
+      printf("\n");
+    } else {
+      printf(" ");
+    }
+  }
+#endif
 }
diff --git a/tests/eval/_ntt/ntt_64_main.c b/tests/eval/_ntt/ntt_64_main.c
@@ -2,7 +2,6 @@
 
 #include <stdio.h>
 #include <assert.h>
-// #define DEBUG
 
 void ntt(const int *array, int l, const int *twiddle, int p, int *dst);