disable nccl in memopt fit a line unittest

7 years ago · 00fbe67ab4
parent 5b635682e6
commit 00fbe67ab4
1 changed files with 1 additions and 1 deletions
--- 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)