From cf0109a5f6674ddf8d381fdf078bccaa152b11e6 Mon Sep 17 00:00:00 2001
From: Jan Cristina <jan.cristina@gmail.com>
Date: Sun, 27 Aug 2023 23:36:18 +0200
Subject: [PATCH] more debug

---
 src/tensor_ops/conv1d/cuda_kernel.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/tensor_ops/conv1d/cuda_kernel.rs b/src/tensor_ops/conv1d/cuda_kernel.rs
index 6fc64a93..fb571300 100644
--- a/src/tensor_ops/conv1d/cuda_kernel.rs
+++ b/src/tensor_ops/conv1d/cuda_kernel.rs
@@ -101,7 +101,7 @@ where
             // RHS (B, G, C/G*K, OL)
             // OUT (B, G, O/G, OL)
             let m = op.chan_out / op.groups;
-            let k = (op.chan_in / op.groups) * op.kernel;
+            let k = op.chan_in * op.kernel;
             let n = op.l_out;
             if op.groups == 1 {
                 // optimizing here for common case