提交 66cc1850 编写于 作者: T tangwei12

add gpu place

上级 fff6f595
...@@ -69,7 +69,9 @@ class TestFillConstantOpWithSelectedRows(OpTest): ...@@ -69,7 +69,9 @@ class TestFillConstantOpWithSelectedRows(OpTest):
def test_fill_constant_with_selected_rows(self): def test_fill_constant_with_selected_rows(self):
places = [core.CPUPlace()] places = [core.CPUPlace()]
# currently only support CPU if core.is_compiled_with_cuda():
places.append(core.CUDAPlace(0))
for place in places: for place in places:
self.check_with_place(place) self.check_with_place(place)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册