Skip to content

Commit dd0b203

Browse files
committed
add note for use mkldnn
1 parent 01f0f16 commit dd0b203

File tree

4 files changed

+5
-5
lines changed

4 files changed

+5
-5
lines changed

paddle/fluid/inference/api/analysis_predictor.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ bool AnalysisPredictor::Init(
7777

7878
OptimizeInferenceProgram();
7979
ctx_ = executor_->Prepare(*inference_program_, 0);
80-
if (config_.use_mkldnn) {
80+
if (config_._use_mkldnn) {
8181
executor_->EnableMKLDNN(*inference_program_);
8282
}
8383

paddle/fluid/inference/api/api_impl.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,7 @@ bool NativePaddlePredictor::Init(
106106
}
107107

108108
ctx_ = executor_->Prepare(*inference_program_, 0);
109-
if (config_.use_mkldnn) {
109+
if (config_._use_mkldnn) {
110110
executor_->EnableMKLDNN(*inference_program_);
111111
}
112112
executor_->CreateVariables(*inference_program_,

paddle/fluid/inference/api/paddle_inference_api.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -121,8 +121,8 @@ struct NativeConfig : public PaddlePredictor::Config {
121121
bool use_gpu{false};
122122
int device{0};
123123
float fraction_of_gpu_memory{-1.f}; // Negative to notify initialization.
124-
// MKLDNN related fields.
125-
bool use_mkldnn{false};
124+
// NOTE: NOT use it, just for the internal test, will discard later
125+
bool _use_mkldnn{false};
126126
// Specify the variable's name of each input.
127127
bool specify_input_name{false};
128128

paddle/fluid/inference/tests/api/analyzer_vis_tester.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ void TestVisualPrediction(bool use_mkldnn) {
7272
cfg.param_file = FLAGS_infer_model + "/__params__";
7373
cfg.prog_file = FLAGS_infer_model + "/__model__";
7474
cfg.use_gpu = false;
75-
cfg.use_mkldnn = use_mkldnn;
75+
cfg._use_mkldnn = use_mkldnn;
7676
cfg.device = 0;
7777
cfg.enable_ir_optim = true;
7878
cfg.ir_passes.push_back("fc_gru_fuse_pass");

0 commit comments

Comments
 (0)