提交 e1d666fb 编写于 作者: A Adam 提交者: Tao Luo

Hotfix to MKL-DNN pool2d tests (#22086)

上级 bf978fff
...@@ -26,6 +26,9 @@ def create_test_mkldnn_use_ceil_class(parent): ...@@ -26,6 +26,9 @@ def create_test_mkldnn_use_ceil_class(parent):
def init_ceil_mode(self): def init_ceil_mode(self):
self.ceil_mode = True self.ceil_mode = True
def init_data_type(self):
self.dtype = np.float32
cls_name = "{0}_{1}".format(parent.__name__, "MKLDNNCeilModeCast") cls_name = "{0}_{1}".format(parent.__name__, "MKLDNNCeilModeCast")
TestMKLDNNPool2DUseCeilCase.__name__ = cls_name TestMKLDNNPool2DUseCeilCase.__name__ = cls_name
globals()[cls_name] = TestMKLDNNPool2DUseCeilCase globals()[cls_name] = TestMKLDNNPool2DUseCeilCase
...@@ -41,6 +44,9 @@ def create_test_mkldnn_class(parent): ...@@ -41,6 +44,9 @@ def create_test_mkldnn_class(parent):
def init_kernel_type(self): def init_kernel_type(self):
self.use_mkldnn = True self.use_mkldnn = True
def init_data_type(self):
self.dtype = np.float32
cls_name = "{0}_{1}".format(parent.__name__, "MKLDNNOp") cls_name = "{0}_{1}".format(parent.__name__, "MKLDNNOp")
TestMKLDNNCase.__name__ = cls_name TestMKLDNNCase.__name__ = cls_name
globals()[cls_name] = TestMKLDNNCase globals()[cls_name] = TestMKLDNNCase
...@@ -78,6 +84,9 @@ class TestAsymPad(TestPool2D_Op): ...@@ -78,6 +84,9 @@ class TestAsymPad(TestPool2D_Op):
def init_global_pool(self): def init_global_pool(self):
self.global_pool = False self.global_pool = False
def init_data_type(self):
self.dtype = np.float32
class TestAsymPadCase1(TestAsymPad): class TestAsymPadCase1(TestAsymPad):
def init_paddings(self): def init_paddings(self):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册