From 16fe5313fd1219e2e7d531ef9b36f64bb557e5e7 Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Thu, 3 Feb 2022 12:59:22 +0900 Subject: [PATCH] wip --- python/tvm/topi/testing/conv2d_backcward_weight_python.py | 2 +- tests/python/relay/test_op_grad_level2.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/python/tvm/topi/testing/conv2d_backcward_weight_python.py b/python/tvm/topi/testing/conv2d_backcward_weight_python.py index 305fb138795c..e8ea2d0367b3 100644 --- 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 index ff053a1c0ad8..acabcef01d08 100644 --- 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)