Skip to content

Commit 7dbdf00

Browse files
committed
remove nccl True in unit test
1 parent 1cce055 commit 7dbdf00

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

python/paddle/fluid/tests/book_memory_optimization/test_memopt_fit_a_line.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,10 +31,10 @@
3131
place = fluid.CPUPlace()
3232
if fluid.core.is_compiled_with_cuda():
3333
device_type = 'CUDA'
34-
use_nccl = True
34+
use_nccl = False
3535
place = fluid.CUDAPlace(0)
3636

37-
places = fluid.layers.get_places(device_count=2, device_type=device_type)
37+
places = fluid.layers.get_places(device_count=0, device_type=device_type)
3838
pd = fluid.layers.ParallelDo(places, use_nccl=use_nccl)
3939
with pd.do():
4040
x_ = pd.read_input(x)

0 commit comments

Comments
 (0)