rust/src/lib.rs: add NTT enums declarations.

supranational · Jan 7, 2024 · a24b42b · a24b42b
1 parent 65739f4
commit a24b42b
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/rust/src/lib.rs b/rust/src/lib.rs
@@ -93,3 +93,23 @@ impl<T> Clone for Gpu_Ptr<T> {
         unsafe { transmute::<_, _>(clone_gpu_ptr_t(transmute::<&_, &_>(self))) }
     }
 }
+
+#[repr(C)]
+pub enum NTTInputOutputOrder {
+    NN = 0,
+    NR = 1,
+    RN = 2,
+    RR = 3,
+}
+
+#[repr(C)]
+pub enum NTTDirection {
+    Forward = 0,
+    Inverse = 1,
+}
+
+#[repr(C)]
+pub enum NTTType {
+    Standard = 0,
+    Coset = 1,
+}