Implemented a simple row-major to col-major problem conversion for TRSM

1 files changed, 10 insertions, 2 deletions

diff --git a/src/routines/level3/xtrsm.hpp b/src/routines/level3/xtrsm.hpp
index b9d5432a..5b42398e 100644
--- a/src/routines/level3/xtrsm.hpp
+++ b/src/routines/level3/xtrsm.hpp
@@ -37,12 +37,20 @@ class Xtrsm: public Xgemm<T> {
   Xtrsm(Queue &queue, EventPointer event, const std::string &name = "TRSM");
 
   // Templated-precision implementation of the routine
-  void DoTrsm(const Layout layout, const Side side, const Triangle triangle,
+  void DoTrsm(const Layout layout, Side side, Triangle triangle,
               const Transpose a_transpose, const Diagonal diagonal,
-              const size_t m, const size_t n,
+              size_t m, size_t n,
               const T alpha,
               const Buffer<T> &a_buffer, const size_t a_offset, const size_t a_ld,
               const Buffer<T> &b_buffer, const size_t b_offset, const size_t b_ld);
+
+  // Implementation of the column-major version
+  void TrsmColMajor(const Side side, const Triangle triangle,
+                    const Transpose a_transpose, const Diagonal diagonal,
+                    const size_t m, const size_t n,
+                    const T alpha,
+                    const Buffer<T> &a_buffer, const size_t a_offset, const size_t a_ld,
+                    const Buffer<T> &b_buffer, const size_t b_offset, const size_t b_ld);
 };
 
 // =================================================================================================