Skip to content
Snippets Groups Projects

Set up cava benchmark

Merged rarbore2 requested to merge cava_opt_3 into main
3 files
+ 12
32
Compare changes
  • Side-by-side
  • Inline
Files
3
+ 11
28
@@ -4,9 +4,8 @@ use std::alloc::{alloc, dealloc, Layout};
use std::marker::PhantomData;
use std::ptr::{copy_nonoverlapping, write_bytes, NonNull};
use std::slice::{from_raw_parts, from_raw_parts_mut};
use std::sync::OnceLock;
use aligned_vec::AVec;
use std::sync::OnceLock;
/*
* Define supporting types, functions, and macros for Hercules RT functions. For
@@ -464,7 +463,7 @@ unsafe impl Sync for __RawPtrSendSync {}
*/
pub struct HerculesImmBox<'a, T> {
#[allow(dead_code)]
cpu_alloc: OnceLock<AVec<T>>,
cpu_alloc: OnceLock<Vec<T>>,
#[cfg(feature = "cuda")]
cuda_alloc: OnceLock<CUDABox>,
@@ -473,32 +472,16 @@ pub struct HerculesImmBox<'a, T> {
cuda_ref: OnceLock<HerculesCUDARef<'a>>,
}
impl<'a, T: Clone> From<&'a [T]> for HerculesImmBox<'a, T> {
impl<'a, T> From<&'a [T]> for HerculesImmBox<'a, T> {
fn from(value: &'a [T]) -> Self {
if value.as_ptr().is_aligned_to(32) {
HerculesImmBox {
cpu_alloc: OnceLock::new(),
#[cfg(feature = "cuda")]
cuda_alloc: OnceLock::new(),
cpu_ref: OnceLock::from(HerculesCPURef::from_slice(value)),
#[cfg(feature = "cuda")]
cuda_ref: OnceLock::new(),
}
} else {
let cpu_alloc = AVec::from_slice(32, value);
let size = value.len() * size_of::<T>();
let cpu_ref =
unsafe { HerculesCPURef::__from_parts(cpu_alloc.as_ptr() as *mut u8, size) };
HerculesImmBox {
cpu_alloc: OnceLock::from(cpu_alloc),
#[cfg(feature = "cuda")]
cuda_alloc: OnceLock::new(),
cpu_ref: OnceLock::from(cpu_ref),
#[cfg(feature = "cuda")]
cuda_ref: OnceLock::new(),
}
HerculesImmBox {
cpu_alloc: OnceLock::new(),
#[cfg(feature = "cuda")]
cuda_alloc: OnceLock::new(),
cpu_ref: OnceLock::from(HerculesCPURef::from_slice(value)),
#[cfg(feature = "cuda")]
cuda_ref: OnceLock::new(),
}
}
}
Loading