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

- Fix to test_conv2d_transpose_mkldnn for GPU

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