From 7dbdf007ecb54598b583ff43d36a08fc21ff008a Mon Sep 17 00:00:00 2001 From: qijun Date: Fri, 2 Mar 2018 11:20:14 +0800 Subject: [PATCH] remove nccl True in unit test --- .../tests/book_memory_optimization/test_memopt_fit_a_line.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/tests/book_memory_optimization/test_memopt_fit_a_line.py b/python/paddle/fluid/tests/book_memory_optimization/test_memopt_fit_a_line.py index a7dd3d91d2..7648bb9fe1 100644 --- a/python/paddle/fluid/tests/book_memory_optimization/test_memopt_fit_a_line.py +++ b/python/paddle/fluid/tests/book_memory_optimization/test_memopt_fit_a_line.py @@ -31,10 +31,10 @@ use_nccl = False place = fluid.CPUPlace() if fluid.core.is_compiled_with_cuda(): device_type = 'CUDA' - use_nccl = True + use_nccl = False place = fluid.CUDAPlace(0) -places = fluid.layers.get_places(device_count=2, device_type=device_type) +places = fluid.layers.get_places(device_count=0, device_type=device_type) pd = fluid.layers.ParallelDo(places, use_nccl=use_nccl) with pd.do(): x_ = pd.read_input(x) -- GitLab