Struct RocmMemoryBackend 

pub struct RocmMemoryBackend { /* private fields */ }

ROCm memory backend implementation

Implementations

impl RocmMemoryBackend

pub fn new(config: RocmConfig) -> Result<Self, RocmError>

Create new ROCm backend

pub fn allocate( &mut self, size: usize, memory_type: RocmMemoryType, ) -> Result<*mut c_void, RocmError>

Allocate device memory

pub fn free( &mut self, ptr: *mut c_void, memory_type: RocmMemoryType, ) -> Result<(), RocmError>

Free device memory

pub fn memcpy( &mut self, dst: *mut c_void, src: *const c_void, size: usize, kind: RocmMemcpyKind, ) -> Result<(), RocmError>

Copy memory

pub fn memcpy_async( &mut self, dst: *mut c_void, src: *const c_void, size: usize, kind: RocmMemcpyKind, stream_id: u32, ) -> Result<(), RocmError>

Asynchronous memory copy

pub fn create_context( &mut self, flags: HipContextFlags, ) -> Result<u32, RocmError>

Create HIP context

pub fn get_device_properties(&self) -> &RocmDeviceProperties

Get device properties

pub fn get_stats(&self) -> &RocmStats

Get statistics

pub fn device_synchronize(&mut self) -> Result<(), RocmError>

Synchronize device

pub fn create_stream(&mut self, priority: Option<i32>) -> Result<u32, RocmError>

Create stream

pub fn destroy_stream(&mut self, stream_id: u32) -> Result<(), RocmError>

Destroy stream

pub fn query_memory_attributes( &self, ptr: *mut c_void, ) -> Result<RocmMemoryAttributes, RocmError>

Query memory attributes

Trait Implementations

impl Send for RocmMemoryBackend

impl Sync for RocmMemoryBackend