From c5b471ce4f1de4b86d61644eec22c9ac85b90bf1 Mon Sep 17 00:00:00 2001
From: "Rabbani, Golam" <golam.rabbani@intel.com>
Date: Fri, 9 Aug 2024 09:47:01 -0700
Subject: [PATCH] fix leading dimensions

---
 cutlass_gemm/cutlass_gemm/cutlass_gemm.hpp | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/cutlass_gemm/cutlass_gemm/cutlass_gemm.hpp b/cutlass_gemm/cutlass_gemm/cutlass_gemm.hpp
index 129b5ff..4066c0f 100644
--- a/cutlass_gemm/cutlass_gemm/cutlass_gemm.hpp
+++ b/cutlass_gemm/cutlass_gemm/cutlass_gemm.hpp
@@ -51,9 +51,9 @@ template<typename DataType, typename OutputType> void cutlass_gemm_wrapper(int M
   float alpha = 1.0f;
   float beta = 0.0f;
 
-  int lda = M;
-  int ldb = K;
-  int ldc = M;
+  int lda = K;
+  int ldb = N;
+  int ldc = N;
 
   Gemm gemm_op;
   gemm_op({