From fd98347610e9acf2dac1d3f3fb06fda25bfba906 Mon Sep 17 00:00:00 2001 From: Zhen Wang Date: Mon, 5 Nov 2018 19:56:25 +0800 Subject: [PATCH] update fuse_statis --- .../tests/api/analyzer_dam_tester.cc | 20 +++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/paddle/fluid/inference/tests/api/analyzer_dam_tester.cc b/paddle/fluid/inference/tests/api/analyzer_dam_tester.cc index 20cb6e31b6a..9fdcf8226be 100644 --- a/paddle/fluid/inference/tests/api/analyzer_dam_tester.cc +++ b/paddle/fluid/inference/tests/api/analyzer_dam_tester.cc @@ -157,7 +157,7 @@ void SetConfig(contrib::AnalysisConfig *cfg) { cfg->use_gpu = false; cfg->device = 0; cfg->specify_input_name = true; - cfg->enable_ir_optim = true; + cfg->enable_ir_optim = false; } void SetInput(std::vector> *inputs) { @@ -198,16 +198,16 @@ TEST(Analyzer_dam, profile) { // Check the fuse status TEST(Analyzer_dam, fuse_statis) { contrib::AnalysisConfig cfg; - // cfg.enable_ir_optim must be set true SetConfig(&cfg); - - int num_ops; - auto predictor = CreatePaddlePredictor(cfg); - auto fuse_statis = GetFuseStatis( - static_cast(predictor.get()), &num_ops); - ASSERT_TRUE(fuse_statis.count("fc_fuse")); - EXPECT_EQ(fuse_statis.at("fc_fuse"), 317); - EXPECT_EQ(num_ops, 2020); + if (cfg.enable_ir_optim) { // cfg.enable_ir_optim must be set true + int num_ops; + auto predictor = CreatePaddlePredictor(cfg); + auto fuse_statis = GetFuseStatis( + static_cast(predictor.get()), &num_ops); + ASSERT_TRUE(fuse_statis.count("fc_fuse")); + EXPECT_EQ(fuse_statis.at("fc_fuse"), 317); + EXPECT_EQ(num_ops, 2020); + } } // Compare result of NativeConfig and AnalysisConfig -- GitLab