huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-18 03:28: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 crate::{CpuStorage, DType, Result, Shape};
use cudarc::driver::CudaSlice; use cudarc::driver::CudaSlice;
pub(crate) type Error = cudarc::driver::DriverError;
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
pub struct CudaDevice(std::sync::Arc<cudarc::driver::CudaDevice>); pub struct CudaDevice(std::sync::Arc<cudarc::driver::CudaDevice>);

2
src/error.rs
View File

@ -35,7 +35,7 @@ pub enum Error {
}, },
#[error(transparent)] #[error(transparent)]
Cudarc(#[from] cudarc::driver::DriverError), Cudarc(#[from] crate::cuda_backend::Error),
} }
pub type Result<T> = std::result::Result<T, Error>; pub type Result<T> = std::result::Result<T, Error>;