Skip to content

P
PaddleDetection

s920243400 / PaddleDetection
与 Fork 源项目一致

Fork自 PaddlePaddle / PaddleDetection

通知 2
Star 0
Fork 0
P
PaddleDetection
提交 87eb8b0e 编写于 作者: M Michal Gallus
浏览文件
操作

Set cpu only for MKL-DNN concat UTs 

test=develop
上级 6fdbb365
隐藏空白更改
内联 并排
Showing with 3 addition and 0 deletion
python/paddle/fluid/tests/unittests/test_concat_mkldnn_op.py
浏览文件 @ 87eb8b0e
...@@ -22,6 +22,7 @@ class TestMKLDNNConcatOp(TestConcatOp): ...@@ -22,6 +22,7 @@ class TestMKLDNNConcatOp(TestConcatOp):
def setUp(self): def setUp(self):
super(TestMKLDNNConcatOp, self).setUp() super(TestMKLDNNConcatOp, self).setUp()
self.attrs["use_mkldnn"] = True self.attrs["use_mkldnn"] = True
self._cpu_only = True
def test_check_grad(self): def test_check_grad(self):
pass pass
...@@ -34,6 +35,7 @@ class TestMKLDNNConcatOp2(TestConcatOp2): ...@@ -34,6 +35,7 @@ class TestMKLDNNConcatOp2(TestConcatOp2):
def setUp(self): def setUp(self):
super(TestMKLDNNConcatOp2, self).setUp() super(TestMKLDNNConcatOp2, self).setUp()
self.attrs["use_mkldnn"] = True self.attrs["use_mkldnn"] = True
self._cpu_only = True
def test_check_grad(self): def test_check_grad(self):
pass pass
...@@ -46,6 +48,7 @@ class TestMKLDNNConcatOp3(TestConcatOp3): ...@@ -46,6 +48,7 @@ class TestMKLDNNConcatOp3(TestConcatOp3):
def setUp(self): def setUp(self):
super(TestMKLDNNConcatOp3, self).setUp() super(TestMKLDNNConcatOp3, self).setUp()
self.attrs["use_mkldnn"] = True self.attrs["use_mkldnn"] = True
self._cpu_only = True
def test_check_grad(self): def test_check_grad(self):
pass pass
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或