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( 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( ldc: out_stride as i32, }; println!( - "lda: {}, ldb {}, ldc: {}, {}, {}, {}", + "FALSE! lda: {}, ldb {}, ldc: {}, {}, {}, {}", cfg.lda, cfg.ldb, cfg.ldc,