AtHeartEngineer/cuda-sumcheck
1
0
Fork 0
mirror of https://github.com/pseXperiments/cuda-sumcheck.git synced 2026-01-09 23:47:57 -05:00
Code Issues Packages Projects Releases Wiki Activity

Impl From<F: PrimeField> trait for FieldBinding

Browse Source
This commit is contained in:
DoHoonKim8
2024-06-17 13:41:00 +00:00
committed by DoHoon Kim
parent 5957edb71c
commit ce95477b5d
1 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
sumcheck/src/lib.rs
View File

@@ -4,6 +4,7 @@ use cudarc::driver::{CudaDevice, LaunchConfig, DeviceRepr, DriverError, LaunchAs
use cudarc::nvrtc::Ptx; use cudarc::nvrtc::Ptx;
use ff::{Field, PrimeField}; use ff::{Field, PrimeField};
use halo2curves::bn256::Fr; use halo2curves::bn256::Fr;
use itertools::Itertools;
include!(concat!(env!("OUT_DIR"), "/bindings.rs")); include!(concat!(env!("OUT_DIR"), "/bindings.rs"));
@@ -17,6 +18,19 @@ impl Default for FieldBinding {
} }
} }
impl<F: PrimeField> From<F> for FieldBinding {
fn from(value: F) -> Self {
let repr = value.to_repr();
let bytes = repr.as_ref();
let data = bytes.chunks(8).map(|bytes| {
u64::from_le_bytes(bytes.try_into().unwrap())
}).collect_vec();
FieldBinding {
data: data.try_into().unwrap()
}
}
}
/// Wrapper struct for APIs using GPU /// Wrapper struct for APIs using GPU
#[derive(Default)] #[derive(Default)]
pub struct GPUApiWrapper<F: Field>(PhantomData<F>); pub struct GPUApiWrapper<F: Field>(PhantomData<F>);