summaryrefslogtreecommitdiff
path: root/src/lib.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/lib.rs')
-rw-r--r--src/lib.rs4
1 files changed, 3 insertions, 1 deletions
diff --git a/src/lib.rs b/src/lib.rs
index 58c2ba52..175d36ad 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,3 +1,4 @@
+mod cpu_backend;
mod device;
mod dtype;
mod error;
@@ -7,10 +8,11 @@ mod storage;
mod strided_index;
mod tensor;
+pub use cpu_backend::CpuStorage;
pub use device::Device;
pub use dtype::{DType, WithDType};
pub use error::{Error, Result};
pub use shape::Shape;
-pub use storage::{CpuStorage, Storage};
+pub use storage::Storage;
use strided_index::StridedIndex;
pub use tensor::{Tensor, TensorId};