From 00fbe67ab4311bce18d0e876487dced4e6da00c5 Mon Sep 17 00:00:00 2001 From: qijun Date: Mon, 5 Mar 2018 20:32:00 +0800 Subject: [PATCH] disable nccl in memopt fit a line unittest --- .../tests/book_memory_optimization/test_memopt_fit_a_line.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 a99d13330b8..7648bb9fe1c 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,7 +31,7 @@ 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=0, device_type=device_type) -- GitLab