add gpu place

66cc1850 · tangwei12 · fff6f595 · 66cc1850
显示空白变更内容
内联并排

Showing with 3 addition and 1 deletion

python/paddle/fluid/tests/unittests/test_fill_constant_op.py python/paddle/fluid/tests/unittests/test_fill_constant_op.py +3 -1

未找到文件。
--- 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)