(&self, count: usize)
| 207 | |
| 208 | #[inline] |
| 209 | fn gpu_alloc(&self, count: usize) -> Result<GpuBuf> { |
| 210 | let mut ptr = std::ptr::null_mut(); |
| 211 | let err = unsafe { (self.ffi.hip_malloc)(&mut ptr, count * 4) }; |
| 212 | if err != 0 { return Err(candle_core::Error::Msg(format!("hipMalloc: {err}"))); } |
| 213 | Ok(GpuBuf { ptr, count, ffi: &*self.ffi as *const RocmFfi }) |
| 214 | } |
| 215 | |
| 216 | #[inline] |
| 217 | fn gpu_upload(&self, data: &[f32]) -> Result<GpuBuf> { |
no outgoing calls
no test coverage detected