Fix an issue caused by uninitialized c in triton_conv2d

voltjia · voltjia · commit 17a7b4779b08 · 2025-01-10T22:49:58.000+08:00
diff --git a/conv2d.py b/conv2d.py
@@ -243,7 +243,7 @@ def triton_conv2d_kernel(
 
 
 def triton_conv2d(input, filter):
-    n, _, h, w = input.shape
+    n, c, h, w = input.shape
     k, _, r, s = filter.shape
     p = h - r + 1
     q = w - s + 1
