提交 1540df51 编写于 作者: J Jacek Czaja

- Fix to test_conv2d_transpose_mkldnn for GPU

test=develop
上级 06d8777e
...@@ -256,6 +256,9 @@ void Conv3DTransposeOpMaker::Make() { ...@@ -256,6 +256,9 @@ void Conv3DTransposeOpMaker::Make() {
"use_cudnn", "use_cudnn",
"(bool, default false) Only used in cudnn kernel, need install cudnn") "(bool, default false) Only used in cudnn kernel, need install cudnn")
.SetDefault(false); .SetDefault(false);
AddAttr<bool>("use_mkldnn",
"(bool, default false) Only used in mkldnn kernel")
.SetDefault(false);
AddAttr<std::string>( AddAttr<std::string>(
"data_format", "data_format",
"(string, default NCHW) Only used in " "(string, default NCHW) Only used in "
......
...@@ -25,6 +25,7 @@ class TestMKLDNN(TestConv2dTransposeOp): ...@@ -25,6 +25,7 @@ class TestMKLDNN(TestConv2dTransposeOp):
self.use_mkldnn = True self.use_mkldnn = True
self.data_format = "NCHW" self.data_format = "NCHW"
self.op_type = "conv2d_transpose" self.op_type = "conv2d_transpose"
self._cpu_only = True
def test_check_grad(self): def test_check_grad(self):
return return
...@@ -42,6 +43,7 @@ class TestMKLDNNWithPad(TestWithPad): ...@@ -42,6 +43,7 @@ class TestMKLDNNWithPad(TestWithPad):
self.use_mkldnn = True self.use_mkldnn = True
self.data_format = "NCHW" self.data_format = "NCHW"
self.op_type = "conv2d_transpose" self.op_type = "conv2d_transpose"
self._cpu_only = True
def test_check_grad(self): def test_check_grad(self):
return return
...@@ -59,6 +61,7 @@ class TestMKLDNNWithStride(TestWithStride): ...@@ -59,6 +61,7 @@ class TestMKLDNNWithStride(TestWithStride):
self.use_mkldnn = True self.use_mkldnn = True
self.data_format = "NCHW" self.data_format = "NCHW"
self.op_type = "conv2d_transpose" self.op_type = "conv2d_transpose"
self._cpu_only = True
def test_check_grad(self): def test_check_grad(self):
return return
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册