convert to col major

ROCm · Jul 11, 2017 · 78955d5 · 78955d5
1 parent bff80c7
commit 78955d5
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/gemm_api.cpp b/src/gemm_api.cpp
@@ -42,6 +42,15 @@ extern "C" miopenStatus_t miopenGemm(miopenHandle_t handle,
                                      void* C,
                                      int ldc)
 {
+
+    // JN make column major
+    if (isDataColMajor == false){
+        std::swap(M, N);
+        std::swap(lda, ldb);
+        std::swap(A, B);
+        isDataColMajor = true;
+    }
+
     return miopen::try_([&] {
         miopen::GemmGeometry gg =
             miopen::CreateMIOpenGemmGeometry(M,