File tree Expand file tree Collapse file tree 5 files changed +8
-5
lines changed Expand file tree Collapse file tree 5 files changed +8
-5
lines changed Original file line number Diff line number Diff line change @@ -18,3 +18,4 @@ cc_test(graph_helper_test SRCS graph_helper_test.cc DEPS graph graph_helper op_r
18
18
cc_test (graph_to_program_pass_test SRCS graph_to_program_pass_test.cc DEPS graph_to_program_pass )
19
19
cc_test (test_graph_pattern_detector SRCS graph_pattern_detector_tester.cc DEPS graph_pattern_detector )
20
20
cc_test (test_fc_fuse_pass SRCS fc_fuse_pass_tester.cc DEPS fc_fuse_pass graph_pattern_detector graph pass graph_traits framework_proto )
21
+ set (GLOB_PASS_LIB fc_fuse_pass attention_lstm_fuse_pass infer_clean_graph_pass fc_lstm_fuse_pass seq_concat_fc_fuse_pass )
Original file line number Diff line number Diff line change @@ -10,7 +10,7 @@ set(FLUID_CORE_MODULES proto_desc memory lod_tensor executor)
10
10
# TODO(panyx0718): Should this be called paddle_fluid_inference_api_internal?
11
11
cc_library (paddle_fluid_api
12
12
SRCS io.cc
13
- DEPS ${FLUID_CORE_MODULES} ${GLOB_OP_LIB} graph_to_program_pass )
13
+ DEPS ${FLUID_CORE_MODULES} ${GLOB_OP_LIB} ${GLOB_PASS_LIB} )
14
14
15
15
get_property (fluid_modules GLOBAL PROPERTY FLUID_MODULES )
16
16
@@ -32,6 +32,7 @@ endif()
32
32
# Create shared library
33
33
cc_library (paddle_fluid_shared SHARED
34
34
SRCS io.cc ${CMAKE_CURRENT_SOURCE_DIR} /api/api.cc ${CMAKE_CURRENT_SOURCE_DIR} /api/api_impl.cc
35
+ ${CMAKE_CURRENT_SOURCE_DIR} /api/analysis_predictor.cc
35
36
DEPS ${fluid_modules} paddle_fluid_api )
36
37
37
38
set_target_properties (paddle_fluid_shared PROPERTIES OUTPUT_NAME paddle_fluid )
Original file line number Diff line number Diff line change @@ -18,10 +18,7 @@ if(APPLE)
18
18
endif (APPLE )
19
19
20
20
21
- set (inference_deps paddle_inference_api paddle_fluid_api analysis pass ir_pass_manager
22
- graph_viz_pass fc_fuse_pass
23
- infer_clean_graph_pass
24
- )
21
+ set (inference_deps paddle_inference_api paddle_fluid_api analysis pass ir_pass_manager ${GLOB_PASS_LIB} )
25
22
26
23
if (WITH_GPU AND TENSORRT_FOUND )
27
24
set (inference_deps ${inference_deps} paddle_inference_tensorrt_subgraph_engine )
Original file line number Diff line number Diff line change @@ -135,5 +135,8 @@ std::unique_ptr<PaddlePredictor> CreatePaddlePredictor<
135
135
} // namespace paddle
136
136
137
137
USE_PASS (fc_fuse_pass);
138
+ USE_PASS (seq_concat_fc_fuse_pass);
139
+ USE_PASS (fc_lstm_fuse_pass);
138
140
USE_PASS (graph_viz_pass);
139
141
USE_PASS (infer_clean_graph_pass);
142
+ USE_PASS (attention_lstm_fuse_pass);
Original file line number Diff line number Diff line change 1
1
{
2
2
global:
3
3
*paddle*;
4
+ *Pass*;
4
5
local:
5
6
*;
6
7
};
You can’t perform that action at this time.
0 commit comments