提交 86523aff 编写于 作者: Q Qiao Longfei

test_sum_op add GPU test

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