Add macro based NTT

Cargo.toml

@@ -15,7 +15,7 @@ members = [
   "tooling/provekit-ffi",
   "tooling/provekit-gnark",
   "tooling/verifier-server",
-  # "ntt",
+  "ntt",
 ]
 exclude = ["playground/passport-input-gen"]
 
@@ -93,7 +93,7 @@ bn254-multiplier-codegen = { package = "provekit-bn254-multiplier-codegen", vers
 fp-rounding = { package = "provekit-fp-rounding", version = "0.1.0" }
 hla = { package = "provekit-hla", version = "0.1.0" }
 skyscraper = { package = "provekit-skyscraper", version = "0.1.0" }
-ntt = { package = "provekit-ntt", version = "0.1.0" }
+provekit-ntt = { path = "ntt" }
 
 # Workspace members - ProveKit
 provekit-bench = { path = "tooling/provekit-bench" }

ntt/Cargo.toml

@@ -9,19 +9,20 @@ license.workspace = true
 homepage.workspace = true
 repository.workspace = true
 
-[lib]
-name = "ntt"
-
 [dependencies]
 ark-bn254.workspace = true
 ark-ff.workspace = true
 rayon.workspace = true
 
 [dev-dependencies]
-proptest.workspace = true
 divan.workspace = true
+ark-std.workspace = true
+
+[target.'cfg(not(target_arch = "wasm32"))'.dev-dependencies]
+proptest.workspace = true
 
 [lints]
+
 workspace = true
 
 [[bench]]

ntt/benches/ntt_engine.rs

@@ -1,6 +1,6 @@
 use divan;
 /// Benchmarks for NTTEngine with divan
-use ntt::{NTTEngine, Pow2};
+use provekit_ntt::NTTEngine;
 
 fn main() {
     // Initialize Rayon thread pool
@@ -14,5 +14,5 @@ const ORDER: usize = 1 << 24;
 /// Benchmark NTTEngine creation with order 1<<24
 #[divan::bench]
 fn create_engine() {
-    let _engine = divan::black_box(NTTEngine::with_order(Pow2::new(ORDER).unwrap()));
+    let _engine = divan::black_box(NTTEngine::with_order(ORDER));
 }

ntt/src/ark_interleaved.rs

@@ -0,0 +1,19 @@
+use {
+    crate::{define_ntt, extend_roots_table},
+    ark_bn254::Fr,
+};
+
+define_ntt!(interleaved_ntt_nr, Fr, ark_kernel);
+
+pub fn ntt_nr_ark(values: &mut [Fr], codeword_size: usize, num_groups: usize) {
+    let new_root = extend_roots_table(codeword_size);
+    interleaved_ntt_nr(new_root.roots(), values, codeword_size, num_groups)
+}
+
+#[inline(always)]
+fn ark_kernel(even: &mut Fr, odd: &mut Fr, omega: &Fr) {
+    let even_value = *even;
+    let odd_twiddled = omega * &*odd;
+    *even = even_value + odd_twiddled;
+    *odd = even_value - odd_twiddled;
+}