Skip to content

Commit 225ecee

Browse files
authored
refine/text classification tester (#13244)
1 parent 6a79ba2 commit 225ecee

File tree

3 files changed

+4
-22
lines changed

3 files changed

+4
-22
lines changed

paddle/fluid/inference/analysis/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -105,6 +105,6 @@ if (NOT EXISTS ${TEXT_CLASSIFICATION_INSTALL_DIR} AND WITH_TESTING AND WITH_INFE
105105
inference_download_and_uncompress(${TEXT_CLASSIFICATION_INSTALL_DIR} ${TEXT_CLASSIFICATION_MODEL_URL} "text-classification-Senta.tar.gz")
106106
endif()
107107

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
109109
EXTRA_DEPS paddle_inference_api paddle_fluid_api analysis_predictor
110110
ARGS --infer_model=${TEXT_CLASSIFICATION_INSTALL_DIR}/text-classification-Senta)

paddle/fluid/inference/analysis/test_text_classification.cc renamed to paddle/fluid/inference/analysis/analyzer_text_classification_tester.cc

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -12,13 +12,14 @@
1212
// See the License for the specific language governing permissions and
1313
// limitations under the License.
1414

15+
#include "paddle/fluid/inference/analysis/analyzer.h"
1516
#include <gflags/gflags.h>
1617
#include <glog/logging.h> // use glog instead of PADDLE_ENFORCE to avoid importing other paddle header files.
1718
#include <gtest/gtest.h>
1819
#include "paddle/fluid/framework/ir/pass.h"
19-
#include "paddle/fluid/inference/analysis/analyzer.h"
2020
#include "paddle/fluid/inference/analysis/ut_helper.h"
2121
#include "paddle/fluid/inference/api/paddle_inference_api.h"
22+
#include "paddle/fluid/inference/api/paddle_inference_pass.h"
2223
#include "paddle/fluid/inference/api/timer.h"
2324

2425
DEFINE_string(infer_model, "", "Directory of the inference model.");
@@ -100,10 +101,3 @@ void Main(int batch_size) {
100101
TEST(text_classification, basic) { Main(FLAGS_batch_size); }
101102

102103
} // 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);

paddle/fluid/inference/api/CMakeLists.txt

Lines changed: 1 addition & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -44,19 +44,7 @@ function(inference_api_test TARGET_NAME)
4444
endfunction(inference_api_test)
4545

4646
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)
6048

6149
cc_test(test_paddle_inference_api
6250
SRCS api_tester.cc

0 commit comments

Comments
 (0)