update ut to meet singleton

PaddlePaddle · Nov 26, 2021 · fee0741 · fee0741 · paddle-bot-old · Nov 26, 2021
1 parent 8b881ac
commit fee0741
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/python/paddle/fluid/tests/unittests/test_fleet_executor.py b/python/paddle/fluid/tests/unittests/test_fleet_executor.py
@@ -32,11 +32,8 @@ def run_fleet_executor(self, place):
         exe.run(empty_program, feed={'x': [1]})
 
     def test_executor_on_single_device(self):
-        places = [fluid.CPUPlace()]
         if fluid.is_compiled_with_cuda():
-            places.append(fluid.CUDAPlace(0))
-        for place in places:
-            self.run_fleet_executor(place)
+            self.run_fleet_executor(fluid.CUDAPlace(0))
 
 
 if __name__ == "__main__":