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

Add a first kernel.

Browse Source
This commit is contained in:
laurent
2023-06-21 20:48:22 +01:00
parent fcb4e6b84f
commit 97d9142dee
3 changed files with 57 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/storage.rs
View File

@ -144,7 +144,10 @@ impl Storage {
let storage = storage.affine_impl(shape, stride, mul, add)?;
Ok(Self::Cpu(storage))
}
Self::Cuda { .. } => todo!(),
Self::Cuda(storage) => {
let storage = storage.affine_impl(shape, stride, mul, add)?;
Ok(Self::Cuda(storage))
}
}
}