set nccl = true

191d8dce · qijun · 7dbdf007 · 191d8dce
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

python/paddle/fluid/tests/book_memory_optimization/test_memopt_fit_a_line.py .../tests/book_memory_optimization/test_memopt_fit_a_line.py +1 -1

未找到文件。
--- 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)