huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 11:08:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add a dummy module.

Browse Source
This commit is contained in:
laurent
2023-06-21 21:16:00 +01:00
parent 97d9142dee
commit 304a557d84
5 changed files with 57 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/cuda_backend.rs
View File

@ -1,7 +1,7 @@
use crate::{CpuStorage, DType, Result, Shape};
use cudarc::driver::{CudaSlice, LaunchAsync, LaunchConfig};
pub(crate) type Error = cudarc::driver::DriverError;
pub type CudaError = cudarc::driver::DriverError;
#[derive(Debug, Clone)]
pub struct CudaDevice(std::sync::Arc<cudarc::driver::CudaDevice>);