test/cpp modify FLAGS_use_mkldnn to FLAGS_use_onednn (PaddlePaddle#74399)

Author: co63oc

test/cpp/inference/api/api_impl_tester.cc

@@ -321,12 +321,12 @@ TEST(inference_api_native, image_classification_gpu) {
 
 #ifdef PADDLE_WITH_DNNL
 TEST(inference_api_native, image_classification_cpu_onednn) {
- FLAGS_use_mkldnn = true;
+ FLAGS_use_onednn = true;
  MainImageClassification(::paddle::PaddlePlace::kCPU);
 }
 
 TEST(inference_api_native, word2vec_cpu_onednn) {
- FLAGS_use_mkldnn = true;
+ FLAGS_use_onednn = true;
  MainWord2Vec(::paddle::PaddlePlace::kCPU);
 }
 #endif

test/cpp/inference/test_helper.h

@@ -26,7 +26,7 @@ limitations under the License. */
 #include "paddle/phi/common/port.h"
 #include "paddle/phi/core/platform/profiler.h"
 
-COMMON_DECLARE_bool(use_mkldnn);
+COMMON_DECLARE_bool(use_onednn);
 
 namespace paddle {
 bool gpu_place_used(const paddle::PaddlePlace& place) {
@@ -235,8 +235,8 @@ void TestInference(
  fetch_targets[fetch_target_names[i]] = cpu_fetches[i];
  }
 
- // 6. If export Flags_use_mkldnn=True, use onednn related ops.
- if (FLAGS_use_mkldnn) executor.EnableONEDNN(*inference_program);
+ // 6. If export Flags_use_onednn=True, use onednn related ops.
+ if (FLAGS_use_onednn) executor.EnableONEDNN(*inference_program);
 
  // 7. Run the inference program
  {