more debug

coreylowman · Aug 26, 2023 · 79ed19c · 79ed19c
1 parent ad222b6
commit 79ed19c
Showing 1 changed file with 10 additions and 1 deletion.
diff --git 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,