Struct MemoryCalculator 

pub struct MemoryCalculator<T: IsFloat + Debug + Copy + Clone> { /* private fields */ }

Wrapper that manages storage of matrices and custom kernels and manages calculation operations.

Implementations

impl<T: IsFloat + Debug + Copy + Clone> MemoryCalculator<T>

pub fn store_matrix(&mut self, matrix: Matrix<T>) -> Result<usize>

Store matrix to calculator and gpu memory.

impl MemoryCalculator<f32>

pub fn init() -> Result<MemoryCalculator<f32>>

Initializes Calculator struct.

pub fn mat_mul( &mut self, left_idx: usize, right_idx: usize, ) -> Result<(Matrix<f32>, usize)>

Multiply Matrix and Matrix.

pub unsafe fn load_custom_fn( &mut self, program_source: &str, kernel_name: &str, parameter_fn: MemoryParameterFunction<f32>, ) -> Result<usize>

Compile and store a custom kernel and build/store a closure to execute said kernel.

pub unsafe fn exec_custom_fn( &mut self, custom_index: usize, input_floats: Option<Vec<f32>>, input_ints: Option<Vec<i32>>, input_mat_idcs: Vec<usize>, ) -> Result<(Matrix<f32>, usize)>

Execute custom kernel via pre-generated closure.

impl MemoryCalculator<f64>

pub fn init() -> Result<MemoryCalculator<f64>>

Initializes Calculator struct.

pub fn mat_mul( &mut self, left_idx: usize, right_idx: usize, ) -> Result<(Matrix<f64>, usize)>

Multiply Matrix and Matrix.

pub unsafe fn load_custom_fn( &mut self, program_source: &str, kernel_name: &str, parameter_fn: MemoryParameterFunction<f64>, ) -> Result<usize>

Compile and store a custom kernel and build/store a closure to execute said kernel.

pub unsafe fn exec_custom_fn( &mut self, custom_index: usize, input_floats: Option<Vec<f64>>, input_ints: Option<Vec<i32>>, input_mat_idcs: Vec<usize>, ) -> Result<(Matrix<f64>, usize)>

Execute custom kernel via pre-generated closure.