From 79ed19c218921c99776d1654d4a39c6ac02fbb68 Mon Sep 17 00:00:00 2001
From: Jan Cristina <jan.cristina@gmail.com>
Date: Sat, 26 Aug 2023 12:06:02 +0200
Subject: [PATCH] more debug

---
 src/tensor_ops/matmul/cuda_kernel.rs | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/tensor_ops/matmul/cuda_kernel.rs b/src/tensor_ops/matmul/cuda_kernel.rs
index 6df144bf..f5eab24c 100644
--- a/src/tensor_ops/matmul/cuda_kernel.rs
+++ b/src/tensor_ops/matmul/cuda_kernel.rs
@@ -40,6 +40,15 @@ fn gemm_cfg<M: Dim, K: Dim, N: Dim, E: Dtype>(
             beta,
             ldc: out_stride as i32,
         };
+        println!(
+            "TRUE! lda: {}, ldb {}, ldc: {}, {}, {}, {}",
+            cfg.lda,
+            cfg.ldb,
+            cfg.ldc,
+            m.size(),
+            k.size(),
+            n.size(),
+        );
         (cfg, true)
     } else {
         // out is stored in column major format
@@ -56,7 +65,7 @@ fn gemm_cfg<M: Dim, K: Dim, N: Dim, E: Dtype>(
             ldc: out_stride as i32,
         };
         println!(
-            "lda: {}, ldb {}, ldc: {}, {}, {}, {}",
+            "FALSE! lda: {}, ldb {}, ldc: {}, {}, {}, {}",
             cfg.lda,
             cfg.ldb,
             cfg.ldc,