Struct CudaDevice 

pub struct CudaDevice { /* private fields */ }

CUDA device manager

Implementations

impl CudaDevice

pub fn new(device_id: u32) -> Result<Self, SimdError>

pub fn get_device_count() -> Result<u32, SimdError>

pub fn get_device_info(device_id: u32) -> Result<GpuDevice, SimdError>

Trait Implementations

impl GpuOperations for CudaDevice

fn allocate<T>(&self, size: usize) -> Result<GpuBuffer<T>, SimdError>

Allocate GPU memory

fn copy_to_device<T>( &self, _host_data: &[T], _gpu_buffer: &mut GpuBuffer<T>, ) -> Result<(), SimdError>

Copy data from host to device

fn copy_to_host<T>( &self, _gpu_buffer: &GpuBuffer<T>, _host_data: &mut [T], ) -> Result<(), SimdError>

Copy data from device to host

fn launch_kernel( &self, _kernel: &str, _config: &KernelConfig, _args: &[&dyn Any], ) -> Result<(), SimdError>

Launch kernel with configuration

fn synchronize(&self) -> Result<(), SimdError>

Synchronize device