fix group_conv3d caculate error (#12500)

apache · Aug 19, 2022 · 9d6039b · 9d6039b
1 parent 41be1b4
commit 9d6039b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/python/tvm/topi/x86/conv3d.py b/python/tvm/topi/x86/conv3d.py
@@ -275,7 +275,7 @@ def _conv3d_ndhwc(cfg, data, kernel, strides, padding, dilation, groups, out_dty
     # pack kernel
     shape = (
         num_filter // oc_bn,
-        in_channel // groups // ic_bn,
+        in_channel // groups // ic_bn if (in_channel // groups // ic_bn) else 1,
         kernel_depth,
         kernel_height,
         kernel_width,
@@ -392,7 +392,7 @@ def _conv3d_ncdhw(cfg, data, kernel, strides, padding, dilation, layout, groups,
     # pack kernel
     shape = (
         num_filter // oc_bn,
-        in_channel // groups // ic_bn,
+        in_channel // groups // ic_bn if (in_channel // groups // ic_bn) else 1,
         kernel_depth,
         kernel_height,
         kernel_width,

diff --git a/tests/python/frontend/onnx/test_forward.py b/tests/python/frontend/onnx/test_forward.py
@@ -2967,7 +2967,7 @@ def repeat(num, dims):
         )
 
     # TODO(jwfromm): Merge with other tests once group_conv3d is supported.
-    for dims in [1, 2]:
+    for dims in [1, 2, 3]:
         # Group Convolution
         verify_conv(
             (1, 8) + repeat(5, dims),