diff --git a/python/paddle/fluid/tests/unittests/mkldnn/test_pool2d_mkldnn_op.py b/python/paddle/fluid/tests/unittests/mkldnn/test_pool2d_mkldnn_op.py index 6de43dd46e5d184ec934f2d85e0c87137e9702e0..feb2a563eeaed7a83a82ec56ec08a0ed8664d126 100644 --- a/python/paddle/fluid/tests/unittests/mkldnn/test_pool2d_mkldnn_op.py +++ b/python/paddle/fluid/tests/unittests/mkldnn/test_pool2d_mkldnn_op.py @@ -18,6 +18,24 @@ import unittest from paddle.fluid.tests.unittests.test_pool2d_op import TestPool2D_Op, TestCase1, TestCase2, TestCase3, TestCase4, TestCase5 +def create_test_mkldnn_use_ceil_class(parent): + class TestMKLDNNPool2DUseCeilCase(parent): + def init_kernel_type(self): + self.use_mkldnn = True + + def init_ceil_mode(self): + self.ceil_mode = True + + cls_name = "{0}_{1}".format(parent.__name__, "MKLDNNCeilModeCast") + TestMKLDNNPool2DUseCeilCase.__name__ = cls_name + globals()[cls_name] = TestMKLDNNPool2DUseCeilCase + + +create_test_mkldnn_use_ceil_class(TestPool2D_Op) +create_test_mkldnn_use_ceil_class(TestCase1) +create_test_mkldnn_use_ceil_class(TestCase2) + + def create_test_mkldnn_class(parent): class TestMKLDNNCase(parent): def init_kernel_type(self):