diff --git a/python/paddle/fluid/tests/unittests/test_fleet_executor.py b/python/paddle/fluid/tests/unittests/test_fleet_executor.py index 6ba9e2d9e21211..76d4a546746a43 100644 --- 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__":