From 3ea7c59f7629701503b206fb5f95eaf3faa5f7ba Mon Sep 17 00:00:00 2001 From: cc <52520497+juncaipeng@users.noreply.github.com> Date: Fri, 3 Apr 2020 14:56:04 +0800 Subject: [PATCH] Set fuse_all_reduce_ops=false for quantization test, test=develop (#23413) --- python/paddle/fluid/contrib/slim/tests/test_quantization_pass.py | 1 + .../fluid/contrib/slim/tests/test_quantization_scale_pass.py | 1 + 2 files changed, 2 insertions(+) diff --git a/python/paddle/fluid/contrib/slim/tests/test_quantization_pass.py b/python/paddle/fluid/contrib/slim/tests/test_quantization_pass.py index eb86b667c0a..dc9b83e4435 100644 --- a/python/paddle/fluid/contrib/slim/tests/test_quantization_pass.py +++ b/python/paddle/fluid/contrib/slim/tests/test_quantization_pass.py @@ -316,6 +316,7 @@ class TestQuantizationFreezePass(unittest.TestCase): build_strategy = fluid.BuildStrategy() build_strategy.memory_optimize = False build_strategy.enable_inplace = False + build_strategy.fuse_all_reduce_ops = False binary = fluid.CompiledProgram(main_graph.graph).with_data_parallel( loss_name=loss.name, build_strategy=build_strategy) quantized_test_program = test_graph.to_program() diff --git a/python/paddle/fluid/contrib/slim/tests/test_quantization_scale_pass.py b/python/paddle/fluid/contrib/slim/tests/test_quantization_scale_pass.py index 0739c9c1f7b..8aa461c438e 100644 --- a/python/paddle/fluid/contrib/slim/tests/test_quantization_scale_pass.py +++ b/python/paddle/fluid/contrib/slim/tests/test_quantization_scale_pass.py @@ -131,6 +131,7 @@ class TestQuantizationScalePass(unittest.TestCase): build_strategy = fluid.BuildStrategy() build_strategy.memory_optimize = False build_strategy.enable_inplace = False + build_strategy.fuse_all_reduce_ops = False binary = fluid.CompiledProgram(main_graph.graph).with_data_parallel( loss_name=loss.name, build_strategy=build_strategy) iters = 5 -- GitLab