@@ -104,14 +104,14 @@ def cuda_places(device_ids=None):
104
104
:code:`FLAGS_selected_gpus=0,1,2`, the returned list would
105
105
be [fluid.CUDAPlace(0), fluid.CUDAPlace(1), fluid.CUDAPlace(2)].
106
106
If :code:`FLAGS_selected_gpus` is not set, all visible
107
- gpu places would be returned.
107
+ gpu places would be returned.
108
108
109
109
If :code:`device_ids` is not None, it should be the device
110
- ids of gpus. For example, if :code:`device_ids=[0,1,2]`,
111
- the returned list would be
110
+ ids of gpus. For example, if :code:`device_ids=[0,1,2]`,
111
+ the returned list would be
112
112
[fluid.CUDAPlace(0), fluid.CUDAPlace(1), fluid.CUDAPlace(2)].
113
-
114
- Args:
113
+
114
+ Args:
115
115
device_ids (None|list(int)|tuple(int)): gpu device id list.
116
116
117
117
Returns:
@@ -133,11 +133,11 @@ def cuda_places(device_ids=None):
133
133
def cpu_places (device_count = None ):
134
134
'''
135
135
Create a list of :code:`fluid.CPUPlace` objects.
136
-
136
+
137
137
If :code:`device_count` is None, the device count would
138
- be determined by environment variable :code:`CPU_NUM`.
138
+ be determined by environment variable :code:`CPU_NUM`.
139
139
If :code:`CPU_NUM` is not set, the device count would
140
- be determined by :code:`multiprocessing.cpu_count()`.
140
+ be determined by :code:`multiprocessing.cpu_count()`.
141
141
142
142
Args:
143
143
device_count (None|int): device number.
@@ -155,9 +155,9 @@ def cuda_pinned_places(device_count=None):
155
155
Create a list of :code:`fluid.CUDAPinnedPlace` objects.
156
156
157
157
If :code:`device_count` is None, the device count would
158
- be determined by environment variable :code:`CPU_NUM`.
158
+ be determined by environment variable :code:`CPU_NUM`.
159
159
If :code:`CPU_NUM` is not set, the device count would
160
- be determined by :code:`multiprocessing.cpu_count()`.
160
+ be determined by :code:`multiprocessing.cpu_count()`.
161
161
162
162
Args:
163
163
device_count (None|int): device number.
0 commit comments