test_sum_op add GPU test

6 years ago · 86523aff25
parent 6253b152e6
commit 86523aff25
1 changed files with 2 additions and 1 deletions
--- a/python/paddle/fluid/tests/unittests/test_sum_op.py
+++ b/python/paddle/fluid/tests/unittests/test_sum_op.py
@ -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)