Move the cpu backend specific bits apart.

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