- Fix to test_conv2d_transpose_mkldnn for GPU

test=develop
f7c96f079b
Jacek Czaja 6 years ago
parent 06d8777ec2
commit 1540df51cf

@ -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

Loading…
Cancel
Save