Skip to content

Commit 3ea19b7

Browse files
committed
fix bug and fc pass ut
1 parent acfdbf0 commit 3ea19b7

File tree

3 files changed

+6
-5
lines changed

3 files changed

+6
-5
lines changed

paddle/fluid/framework/ir/graph_pattern_detector.cc

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -536,22 +536,21 @@ PDNode* patterns::FC(PDPattern* pattern, const std::string& name_scope,
536536
mul_out_var = pattern->NewNode(name_scope, "mul_out")
537537
->AsIntermediate()
538538
->assert_is_only_output_of_op("mul")
539-
->assert_is_op_input("elementwise_add", "X");
539+
->assert_is_op_input("elementwise_add");
540540
// bias
541541
bias = pattern->NewNode(name_scope, "fc_bias")
542542
->AsInput()
543-
->assert_is_persistable_var()
544-
->assert_is_op_input("elementwise_add", "Y");
543+
->assert_is_op_input("elementwise_add");
545544
// output
546545
fc_out = pattern->NewNode(name_scope, "fc_out")
547546
->AsOutput()
548-
->assert_is_op_output("elementwise_add", "Out");
547+
->assert_is_op_output("elementwise_add");
549548
mul_op->LinksFrom({x, mul_weight_var}).LinksTo({mul_out_var});
550549
elementwise_add_op->LinksFrom({mul_out_var, bias}).LinksTo({fc_out});
551550
} else {
552551
fc_out = pattern->NewNode(name_scope, "fc_out")
553552
->AsOutput()
554-
->assert_is_op_output("mul", "Out");
553+
->assert_is_op_output("mul");
555554
mul_op->LinksFrom({mul_weight_var, x}).LinksTo({fc_out});
556555
}
557556
return fc_out;

paddle/fluid/inference/analysis/analyzer_lac_tester.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
#include "paddle/fluid/inference/analysis/analyzer.h"
1616
#include <gtest/gtest.h>
1717
#include "paddle/fluid/framework/ir/fuse_pass_base.h"
18+
#include "paddle/fluid/inference/analysis/ut_helper.h"
1819
#include "paddle/fluid/inference/api/analysis_predictor.h"
1920
#include "paddle/fluid/inference/api/helper.h"
2021
#include "paddle/fluid/inference/api/paddle_inference_pass.h"

paddle/fluid/inference/analysis/analyzer_ner_tester.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
#include "paddle/fluid/inference/analysis/analyzer.h"
1616
#include <gtest/gtest.h>
1717
#include "paddle/fluid/framework/ir/fuse_pass_base.h"
18+
#include "paddle/fluid/inference/analysis/ut_helper.h"
1819
#include "paddle/fluid/inference/api/analysis_predictor.h"
1920
#include "paddle/fluid/inference/api/helper.h"
2021
#include "paddle/fluid/inference/api/paddle_inference_pass.h"

0 commit comments

Comments
 (0)