poc/ntt-cuda: modernize.

argumentcomputer · Jan 7, 2024 · ada2ea2 · ada2ea2
1 parent a24b42b
commit ada2ea2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 23 deletions.
diff --git a/poc/ntt-cuda/src/lib.rs b/poc/ntt-cuda/src/lib.rs
@@ -2,27 +2,7 @@
 // Licensed under the Apache License, Version 2.0, see LICENSE for details.
 // SPDX-License-Identifier: Apache-2.0
 
-sppark::cuda_error!();
-
-#[repr(C)]
-pub enum NTTInputOutputOrder {
-    NN = 0,
-    NR = 1,
-    RN = 2,
-    RR = 3,
-}
-
-#[repr(C)]
-enum NTTDirection {
-    Forward = 0,
-    Inverse = 1,
-}
-
-#[repr(C)]
-enum NTTType {
-    Standard = 0,
-    Coset = 1,
-}
+use sppark::{NTTInputOutputOrder, NTTDirection, NTTType};
 
 extern "C" {
     fn compute_ntt(
@@ -32,7 +12,7 @@ extern "C" {
         ntt_order: NTTInputOutputOrder,
         ntt_direction: NTTDirection,
         ntt_type: NTTType,
-    ) -> cuda::Error;
+    ) -> sppark::Error;
 }
 
 /// Compute an in-place NTT on the input data.

diff --git a/poc/ntt-cuda/tests/ntt.rs b/poc/ntt-cuda/tests/ntt.rs
@@ -2,7 +2,7 @@
 // Licensed under the Apache License, Version 2.0, see LICENSE for details.
 // SPDX-License-Identifier: Apache-2.0
 
-use ntt_cuda::NTTInputOutputOrder;
+use sppark::NTTInputOutputOrder;
 
 const DEFAULT_GPU: usize = 0;