File tree Expand file tree Collapse file tree 3 files changed +4
-22
lines changed Expand file tree Collapse file tree 3 files changed +4
-22
lines changed Original file line number Diff line number Diff line change @@ -105,6 +105,6 @@ if (NOT EXISTS ${TEXT_CLASSIFICATION_INSTALL_DIR} AND WITH_TESTING AND WITH_INFE
105
105
inference_download_and_uncompress (${TEXT_CLASSIFICATION_INSTALL_DIR} ${TEXT_CLASSIFICATION_MODEL_URL} "text-classification-Senta.tar.gz" )
106
106
endif ()
107
107
108
- inference_analysis_test (test_text_classification SRCS test_text_classification .cc
108
+ inference_analysis_test (test_text_classification SRCS analyzer_text_classification_tester .cc
109
109
EXTRA_DEPS paddle_inference_api paddle_fluid_api analysis_predictor
110
110
ARGS --infer_model=${TEXT_CLASSIFICATION_INSTALL_DIR}/text-classification-Senta )
Original file line number Diff line number Diff line change 12
12
// See the License for the specific language governing permissions and
13
13
// limitations under the License.
14
14
15
+ #include " paddle/fluid/inference/analysis/analyzer.h"
15
16
#include < gflags/gflags.h>
16
17
#include < glog/logging.h> // use glog instead of PADDLE_ENFORCE to avoid importing other paddle header files.
17
18
#include < gtest/gtest.h>
18
19
#include " paddle/fluid/framework/ir/pass.h"
19
- #include " paddle/fluid/inference/analysis/analyzer.h"
20
20
#include " paddle/fluid/inference/analysis/ut_helper.h"
21
21
#include " paddle/fluid/inference/api/paddle_inference_api.h"
22
+ #include " paddle/fluid/inference/api/paddle_inference_pass.h"
22
23
#include " paddle/fluid/inference/api/timer.h"
23
24
24
25
DEFINE_string (infer_model, " " , " Directory of the inference model." );
@@ -100,10 +101,3 @@ void Main(int batch_size) {
100
101
TEST (text_classification, basic) { Main (FLAGS_batch_size); }
101
102
102
103
} // namespace paddle
103
-
104
- USE_PASS (fc_fuse_pass);
105
- USE_PASS (seq_concat_fc_fuse_pass);
106
- USE_PASS (fc_lstm_fuse_pass);
107
- USE_PASS (graph_viz_pass);
108
- USE_PASS (infer_clean_graph_pass);
109
- USE_PASS (attention_lstm_fuse_pass);
Original file line number Diff line number Diff line change @@ -44,19 +44,7 @@ function(inference_api_test TARGET_NAME)
44
44
endfunction (inference_api_test )
45
45
46
46
cc_library (paddle_inference_api SRCS api.cc api_impl.cc helper.cc DEPS lod_tensor )
47
- cc_library (analysis_predictor SRCS analysis_predictor.cc DEPS paddle_inference_api
48
- analysis
49
- ir_pass_manager
50
- pass
51
- fc_fuse_pass
52
- fc_lstm_fuse_pass
53
- seq_concat_fc_fuse_pass
54
- graph_viz_pass
55
- infer_clean_graph_pass
56
- graph_pattern_detector
57
- infer_clean_graph_pass
58
- attention_lstm_fuse_pass
59
- )
47
+ cc_library (analysis_predictor SRCS analysis_predictor.cc DEPS paddle_inference_api analysis )
60
48
61
49
cc_test (test_paddle_inference_api
62
50
SRCS api_tester.cc
You can’t perform that action at this time.
0 commit comments