wip

masahi · Feb 4, 2022 · 16fe531 · 16fe531
1 parent 2167c25
commit 16fe531
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/python/tvm/topi/testing/conv2d_backcward_weight_python.py b/python/tvm/topi/testing/conv2d_backcward_weight_python.py
@@ -71,7 +71,7 @@ def conv2d_backward_weight_nchw_python(
                         for p in range(P):
                             for q in range(Q):
                                 if not is_depth_wise:
-                                    in_c = c
+                                    in_c = k
                                 else:
                                     in_c = k // channel_mult
 

diff --git a/tests/python/relay/test_op_grad_level2.py b/tests/python/relay/test_op_grad_level2.py
@@ -269,10 +269,10 @@ def verify_conv2d_backward_weight(
 def test_conv2d_backward_weight():
     # verify_conv2d_backward_weight((2, 8, 32, 32), (2, 4, 32, 32), (3, 3), (1, 1), (1, 1))
     # verify_conv2d_backward_weight((2, 16, 15, 15), (2, 3, 32, 32), (3, 3), (2, 2), (0, 0))
-    verify_conv2d_backward_weight((1, 16, 32, 32), (1, 16, 32, 32), (3, 3), (1, 1), (1, 1), groups=16, out_channels=16)
-    # verify_conv2d_backward_weight(
-    #     (1, 32, 32, 32), (1, 16, 32, 32), (3, 3), (1, 1), (1, 1), groups=8
-    # )
+    # verify_conv2d_backward_weight((1, 16, 32, 32), (1, 16, 32, 32), (3, 3), (1, 1), (1, 1), groups=16, out_channels=16)
+    verify_conv2d_backward_weight(
+        (1, 32, 32, 32), (1, 16, 32, 32), (3, 3), (1, 1), (1, 1), groups=8
+    )
     # verify_conv2d_backward_weight((1, 32, 32, 32), (1, 16, 32, 32), (3, 3), (1, 1), (1, 1), groups=16, out_channels=32)