diff --git a/paddle/fluid/framework/details/reference_count_pass.cc b/paddle/fluid/framework/details/reference_count_pass.cc index 0c3d8d5caec0015c5696223db2e4b75a8d79e5e1..c218e55b70dc7c5e140a8611b3b8310bc43b2c0e 100644 --- a/paddle/fluid/framework/details/reference_count_pass.cc +++ b/paddle/fluid/framework/details/reference_count_pass.cc @@ -335,6 +335,7 @@ std::unique_ptr ReferenceCountPass::ApplyImpl( var_name); ref_cnts[i].emplace(var_name, result.size()); last_live_ops_of_vars[i].emplace(var_name, std::move(result)); + break; } // Seldomly, all preceding trying failed. diff --git a/python/paddle/fluid/framework.py b/python/paddle/fluid/framework.py index 7abd2a23aad34c76ce1544e64703bc8bde57e007..be8008cabd08de0cf957d93ca71ac38fc592af79 100644 --- a/python/paddle/fluid/framework.py +++ b/python/paddle/fluid/framework.py @@ -104,14 +104,14 @@ def cuda_places(device_ids=None): :code:`FLAGS_selected_gpus=0,1,2`, the returned list would be [fluid.CUDAPlace(0), fluid.CUDAPlace(1), fluid.CUDAPlace(2)]. If :code:`FLAGS_selected_gpus` is not set, all visible - gpu places would be returned. + gpu places would be returned. If :code:`device_ids` is not None, it should be the device - ids of gpus. For example, if :code:`device_ids=[0,1,2]`, - the returned list would be + ids of gpus. For example, if :code:`device_ids=[0,1,2]`, + the returned list would be [fluid.CUDAPlace(0), fluid.CUDAPlace(1), fluid.CUDAPlace(2)]. - - Args: + + Args: device_ids (None|list(int)|tuple(int)): gpu device id list. Returns: @@ -133,11 +133,11 @@ def cuda_places(device_ids=None): def cpu_places(device_count=None): ''' Create a list of :code:`fluid.CPUPlace` objects. - + If :code:`device_count` is None, the device count would - be determined by environment variable :code:`CPU_NUM`. + be determined by environment variable :code:`CPU_NUM`. If :code:`CPU_NUM` is not set, the device count would - be determined by :code:`multiprocessing.cpu_count()`. + be determined by :code:`multiprocessing.cpu_count()`. Args: device_count (None|int): device number. @@ -155,9 +155,9 @@ def cuda_pinned_places(device_count=None): Create a list of :code:`fluid.CUDAPinnedPlace` objects. If :code:`device_count` is None, the device count would - be determined by environment variable :code:`CPU_NUM`. + be determined by environment variable :code:`CPU_NUM`. If :code:`CPU_NUM` is not set, the device count would - be determined by :code:`multiprocessing.cpu_count()`. + be determined by :code:`multiprocessing.cpu_count()`. Args: device_count (None|int): device number.