Set cpu only for MKL-DNN concat UTs

test=develop
ce_debug
Michal Gallus 6 years ago
parent 6fdbb365ce
commit 87eb8b0e28

@ -22,6 +22,7 @@ class TestMKLDNNConcatOp(TestConcatOp):
def setUp(self):
super(TestMKLDNNConcatOp, self).setUp()
self.attrs["use_mkldnn"] = True
self._cpu_only = True
def test_check_grad(self):
pass
@ -34,6 +35,7 @@ class TestMKLDNNConcatOp2(TestConcatOp2):
def setUp(self):
super(TestMKLDNNConcatOp2, self).setUp()
self.attrs["use_mkldnn"] = True
self._cpu_only = True
def test_check_grad(self):
pass
@ -46,6 +48,7 @@ class TestMKLDNNConcatOp3(TestConcatOp3):
def setUp(self):
super(TestMKLDNNConcatOp3, self).setUp()
self.attrs["use_mkldnn"] = True
self._cpu_only = True
def test_check_grad(self):
pass

Loading…
Cancel
Save