Struct RocmBackend 

pub struct RocmBackend { /* private fields */ }

ROCm backend implementation

Implementations

impl RocmBackend

pub fn new() -> Result<Self, BackendError>

pub fn is_available() -> bool

Trait Implementations

impl Backend for RocmBackend

fn backend_type(&self) -> GpuBackend

Get backend type

fn initialize(&mut self) -> Result<(), BackendError>

Initialize the backend

fn device_count(&self) -> Result<u32, BackendError>

Get device count

fn device_capabilities( &self, _device_id: u32, ) -> Result<DeviceCapabilities, BackendError>

Get device capabilities

fn set_device(&mut self, device_id: u32) -> Result<(), BackendError>

Set active device

fn allocate(&self, size: usize) -> Result<DeviceMemory, BackendError>

Allocate memory on device

fn deallocate(&self, _memory: DeviceMemory) -> Result<(), BackendError>

Free device memory

fn copy_to_device( &self, _src: &[u8], _dst: &DeviceMemory, ) -> Result<(), BackendError>

Copy memory from host to device

fn copy_to_host( &self, _src: &DeviceMemory, _dst: &mut [u8], ) -> Result<(), BackendError>

Copy memory from device to host

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

Synchronize device

fn launch_kernel( &self, _kernel: &CompiledKernel, _args: &[KernelArg], ) -> Result<(), BackendError>

Launch kernel