From c986a560f4b2d0cf0b7aeda5c639f0174127eda5 Mon Sep 17 00:00:00 2001
From: nathaniel <nathaniel.simard.42@gmail.com>
Date: Fri, 19 Jul 2024 09:50:51 -0400
Subject: [PATCH] Fix layout public

---
 crates/cubecl-linalg/src/tensor/layout.rs | 1 +
 crates/cubecl-linalg/src/tensor/mod.rs    | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/crates/cubecl-linalg/src/tensor/layout.rs b/crates/cubecl-linalg/src/tensor/layout.rs
index 0b4e0b0a..03a4962f 100644
--- 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
index 3a474b8b..c06bc855 100644
--- 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::*;