Fix layout public

tracel-ai · Jul 19, 2024 · c986a56 · c986a56
1 parent 50295f2
commit c986a56
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/crates/cubecl-linalg/src/tensor/layout.rs b/crates/cubecl-linalg/src/tensor/layout.rs
@@ -15,6 +15,7 @@ pub enum MatrixLayout {
     HighlyPermuted,
 }
 
+/// Return the layout of a matrix given the strides.
 pub fn matrix_layout(strides: &[usize]) -> MatrixLayout {
     let rank = strides.len();
     if rank <= 1 {

diff --git a/crates/cubecl-linalg/src/tensor/mod.rs b/crates/cubecl-linalg/src/tensor/mod.rs
@@ -1,4 +1,4 @@
 mod base;
 mod layout;
 pub use base::*;
-pub(crate) use layout::*;
+pub use layout::*;