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 the cublas handle to the cuda device.

Browse Source
This commit is contained in:
laurent
2023-06-22 18:03:53 +01:00
parent 7d9a8ff3f9
commit 683730c21d
3 changed files with 114 additions and 73 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/storage.rs
View File

@ -22,7 +22,7 @@ impl Storage {
pub fn device(&self) -> Device {
match self {
Self::Cpu(_) => Device::Cpu,
Self::Cuda(storage) => Device::Cuda(storage.device()),
Self::Cuda(storage) => Device::Cuda(storage.device().clone()),
}
}