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

Simplify the pattern matching logic in the cuda backend.

Browse Source
This commit is contained in:
laurent
2023-06-29 09:21:11 +01:00
parent eda46d2df2
commit 122e334d0c
5 changed files with 78 additions and 89 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-core/src/storage.rs
View File

@ -9,11 +9,11 @@ pub enum Storage {
}
impl Storage {
pub fn try_clone(&self) -> Result<Self> {
pub fn try_clone(&self, layout: &Layout) -> Result<Self> {
match self {
Self::Cpu(storage) => Ok(Self::Cpu(storage.clone())),
Self::Cuda(storage) => {
let storage = storage.try_clone()?;
let storage = storage.try_clone(layout)?;
Ok(Self::Cuda(storage))
}
}