From 191d8dcec1784838bda274e2575a397b1cb42994 Mon Sep 17 00:00:00 2001 From: qijun Date: Fri, 2 Mar 2018 11:25:13 +0800 Subject: [PATCH] set nccl = true --- .../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 7648bb9fe1c..a99d13330b8 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 = False + use_nccl = True place = fluid.CUDAPlace(0) places = fluid.layers.get_places(device_count=0, device_type=device_type) -- GitLab