huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 19:18:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use a type alias for cuda errors.

Browse Source
This commit is contained in:
laurent
2023-06-21 19:50:00 +01:00
parent 71735c7a02
commit deb6091099
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/cuda_backend.rs
View File

@ -1,6 +1,8 @@
use crate::{CpuStorage, DType, Result, Shape};
use cudarc::driver::CudaSlice;
pub(crate) type Error = cudarc::driver::DriverError;
#[derive(Debug, Clone)]
pub struct CudaDevice(std::sync::Arc<cudarc::driver::CudaDevice>);