From 87eb8b0e28a4b95d8427bd66cbff33f107ad069e Mon Sep 17 00:00:00 2001 From: Michal Gallus Date: Tue, 4 Dec 2018 14:44:01 +0100 Subject: [PATCH] Set cpu only for MKL-DNN concat UTs test=develop --- python/paddle/fluid/tests/unittests/test_concat_mkldnn_op.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/python/paddle/fluid/tests/unittests/test_concat_mkldnn_op.py b/python/paddle/fluid/tests/unittests/test_concat_mkldnn_op.py index 0ea44c0e4e4..0f2130f9049 100644 --- a/python/paddle/fluid/tests/unittests/test_concat_mkldnn_op.py +++ b/python/paddle/fluid/tests/unittests/test_concat_mkldnn_op.py @@ -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 -- GitLab