add gpu place

7 years ago · 66cc1850a8
parent fff6f595ff
commit 66cc1850a8
1 changed files with 3 additions and 1 deletions
--- a/python/paddle/fluid/tests/unittests/test_fill_constant_op.py
+++ b/python/paddle/fluid/tests/unittests/test_fill_constant_op.py
@ -69,7 +69,9 @@ class TestFillConstantOpWithSelectedRows(OpTest):

    def test_fill_constant_with_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:
            self.check_with_place(place)