Skip to content

Commit 54f21a5

Browse files
committed
Add test for ceil mode
test=develop
1 parent 7d8f639 commit 54f21a5

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

python/paddle/fluid/tests/unittests/mkldnn/test_pool2d_mkldnn_op.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,24 @@
1818
from paddle.fluid.tests.unittests.test_pool2d_op import TestPool2D_Op, TestCase1, TestCase2, TestCase3, TestCase4, TestCase5
1919

2020

21+
def create_test_mkldnn_use_ceil_class(parent):
22+
class TestMKLDNNPool2DUseCeilCase(parent):
23+
def init_kernel_type(self):
24+
self.use_mkldnn = True
25+
26+
def init_ceil_mode(self):
27+
self.ceil_mode = True
28+
29+
cls_name = "{0}_{1}".format(parent.__name__, "MKLDNNCeilModeCast")
30+
TestMKLDNNPool2DUseCeilCase.__name__ = cls_name
31+
globals()[cls_name] = TestMKLDNNPool2DUseCeilCase
32+
33+
34+
create_test_mkldnn_use_ceil_class(TestPool2D_Op)
35+
create_test_mkldnn_use_ceil_class(TestCase1)
36+
create_test_mkldnn_use_ceil_class(TestCase2)
37+
38+
2139
def create_test_mkldnn_class(parent):
2240
class TestMKLDNNCase(parent):
2341
def init_kernel_type(self):

0 commit comments

Comments
 (0)