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 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

9
src/lib.rs
View File

@ -1,7 +1,9 @@
mod cpu_backend;
#[cfg(feature = "cuda")]
mod cuda_backend;
mod device;
mod dtype;
mod dummy_cuda_backend;
mod error;
mod op;
mod shape;
@ -10,7 +12,6 @@ mod strided_index;
mod tensor;
pub use cpu_backend::CpuStorage;
pub use cuda_backend::{CudaDevice, CudaStorage};
pub use device::{Device, DeviceLocation};
pub use dtype::{DType, WithDType};
pub use error::{Error, Result};
@ -18,3 +19,9 @@ pub use shape::Shape;
pub use storage::Storage;
use strided_index::StridedIndex;
pub use tensor::{Tensor, TensorId};
#[cfg(feature = "cuda")]
pub use cuda_backend::{CudaDevice, CudaError, CudaStorage};
#[cfg(not(feature = "cuda"))]
pub use dummy_cuda_backend::{CudaDevice, CudaError, CudaStorage};