Fix ntt-1024 test

Fixes #5
slothy-optimizer · Jul 23, 2024 · 7a9b285 · 7a9b285
1 parent a4a6d87
commit 7a9b285
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/Makefile b/Makefile
@@ -15,7 +15,7 @@ include tests/ntt-256/ntt-256.mk
 include tests/ntt-384/ntt-384.mk
 include tests/ntt-512/ntt-512.mk
 include tests/ntt-768/ntt-768.mk
-#include tests/ntt-1024/ntt-1024.mk
+include tests/ntt-1024/ntt-1024.mk
 include tests/ntt-n256/ntt-n256.mk
 include tests/ntt-dilithium/ntt-dilithium.mk
 include tests/ntt-kyber/ntt-kyber.mk

diff --git a/tests/ntt-1024/main.c b/tests/ntt-1024/main.c
@@ -833,11 +833,15 @@ int run_test_ntt_incomplete_double_rev4()
 
 void poly_mul_1024( int32_t *dst, int32_t *srcA, int32_t *srcB )
 {
+    int32_t params[] = {
+        modulus, modulus_inv_u32
+    };
+
     int32_t tmpB[2*NTT_SIZE];
     ntt_u32_mve_rev4( srcA );
     ntt_u32_mve_double_rev4( srcB, tmpB );
-    twisted_cyclic_mul_deg4_u32_add_sub_rev_mve( srcA, tmpB, dst );
-    //ntt_u32_mve_bitrev_skipfirst( dst );
+    twisted_cyclic_mul_deg4_u32_add_sub_rev_mve( srcA, tmpB, dst, params );
+    ntt_u32_mve_bitrev_skipfirst( dst );
 }
 
 int run_test_poly_mul()
@@ -937,5 +941,6 @@ int main(void)
         return( 1 );
 #endif /* TEST_NTT_DOUBLE */
 
+    debug_printf( "ALL GOOD!\n" );
     return( ret );
 }