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

Add some specific errors rather than panicking.

Browse Source
This commit is contained in:
laurent
2023-06-22 07:51:53 +01:00
parent db35b31050
commit fc26bab3ed
3 changed files with 14 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/cuda_backend.rs
View File

@ -1,4 +1,4 @@
use crate::{CpuStorage, DType, Result, Shape};
use crate::{CpuStorage, DType, Error, Result, Shape};
use cudarc::driver::{CudaSlice, LaunchAsync, LaunchConfig};
pub type CudaError = cudarc::driver::DriverError;
@ -92,7 +92,7 @@ impl CudaStorage {
match self {
Self::F32(arg) => {
if !shape.is_contiguous(stride) {
todo!("affine is only implemented for the contiguous case")
return Err(Error::RequiresContiguous { op: "affine" });
}
let dev = arg.device();
let module_name = "affine_f32";