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

Malloc the correct amount of memory

Browse Source
This commit is contained in:
DoHoonKim8
2024-09-23 10:05:35 +00:00
parent 57bc9ac507
commit bc8340d1d2
2 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
sumcheck/src/gpu/sumcheck.rs
View File

@@ -1,8 +1,7 @@
use std::cell::{RefCell, RefMut};
use cudarc::driver::{CudaSlice, CudaView, CudaViewMut, DriverError, LaunchAsync, LaunchConfig};
use cudarc::driver::{CudaView, CudaViewMut, DriverError, LaunchAsync, LaunchConfig};
use ff::PrimeField;
use itertools::Itertools;
use crate::{
fieldbinding::{FromFieldBinding, ToFieldBinding},

2
sumcheck/src/lib.rs
View File

@@ -72,7 +72,7 @@ impl<F: PrimeField + FromFieldBinding<F> + ToFieldBinding<F>> GPUApiWrapper<F> {
}
pub fn malloc_on_device(&self, len: usize) -> Result<CudaSlice<FieldBinding>, DriverError> {
let device_ptr = unsafe { self.gpu.alloc(len << 5)? };
let device_ptr = unsafe { self.gpu.alloc::<FieldBinding>(len)? };
Ok(device_ptr)
}