update depthwise conv2d ut (PaddlePaddle#821)

graphcore · Jun 21, 2022 · 04be283 · 04be283
1 parent 57d7999
commit 04be283
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/python/paddle/fluid/tests/unittests/ipu/test_conv2d_transpose_op_ipu.py b/python/paddle/fluid/tests/unittests/ipu/test_conv2d_transpose_op_ipu.py
@@ -150,5 +150,13 @@ def set_op_attrs(self):
  self.attrs['output_size'] = [12, 12]
 
 
+class TestCase11(TestBase):
+
+ # Depthwise conv2d transpose
+ def set_op_attrs(self):
+ super().set_op_attrs()
+ self.attrs['groups'] = 3
+
+
 if __name__ == "__main__":
  unittest.main()
diff --git a/python/paddle/fluid/tests/unittests/ipu/test_conv_op_ipu.py b/python/paddle/fluid/tests/unittests/ipu/test_conv_op_ipu.py
@@ -108,7 +108,7 @@ def set_op_attrs(self):
 
 
 class TestCase5(TestBase):
-
+ # Depthwise conv2d
  def set_op_attrs(self):
  super().set_op_attrs()
  self.attrs['groups'] = 3