Fix an issue caused by uninitialized c in triton_conv2d

InfiniTensor · Jan 10, 2025 · 17a7b47 · 17a7b47
1 parent d421777
commit 17a7b47
Showing 1 changed file with 1 addition and 1 deletion.
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