We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1cce055 commit 7dbdf00Copy full SHA for 7dbdf00
python/paddle/fluid/tests/book_memory_optimization/test_memopt_fit_a_line.py
@@ -31,10 +31,10 @@
31
place = fluid.CPUPlace()
32
if fluid.core.is_compiled_with_cuda():
33
device_type = 'CUDA'
34
- use_nccl = True
+ use_nccl = False
35
place = fluid.CUDAPlace(0)
36
37
-places = fluid.layers.get_places(device_count=2, device_type=device_type)
+places = fluid.layers.get_places(device_count=0, device_type=device_type)
38
pd = fluid.layers.ParallelDo(places, use_nccl=use_nccl)
39
with pd.do():
40
x_ = pd.read_input(x)
0 commit comments