Skip to content

Commit 3bfdd75

Browse files
authored
Replace EnableMKLDNN EnableONEDNN [fluid_ops] (#74180)
1 parent 4b3b856 commit 3bfdd75

40 files changed

+61
-61
lines changed

paddle/fluid/framework/executor.cc

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -184,7 +184,7 @@ void Executor::Run(const ProgramDesc& pdesc,
184184
phi::RecordEvent record_run(
185185
"Executor::Run", phi::TracerEventType::UserDefined, 1);
186186
platform::RecordBlock b(block_id);
187-
if (FLAGS_use_mkldnn) EnableMKLDNN(pdesc);
187+
if (FLAGS_use_mkldnn) EnableONEDNN(pdesc);
188188
auto ctx = Prepare(pdesc, block_id, skip_ref_cnt_vars, force_disable_gc);
189189
#ifdef PADDLE_WITH_DNNL
190190
platform::AttachPointerHashToONEDNNKey(this, place_);
@@ -330,7 +330,7 @@ void Executor::Run(const ProgramDesc& program,
330330
phi::RecordEvent record_run(
331331
"Executor::Run", phi::TracerEventType::UserDefined, 1);
332332
platform::RecordBlock b(kProgramId);
333-
if (FLAGS_use_mkldnn) EnableMKLDNN(program);
333+
if (FLAGS_use_mkldnn) EnableONEDNN(program);
334334
#ifdef PADDLE_WITH_DNNL
335335
platform::AttachPointerHashToONEDNNKey(this, place_);
336336
#endif
@@ -592,7 +592,7 @@ void Executor::RunPreparedContext(
592592
}
593593
}
594594

595-
void Executor::EnableMKLDNN(const ProgramDesc& program) {
595+
void Executor::EnableONEDNN(const ProgramDesc& program) {
596596
#ifdef PADDLE_WITH_DNNL
597597
VLOG(3) << "use_mkldnn=True";
598598
for (size_t bid = 0; bid < program.Size(); ++bid) {

paddle/fluid/framework/executor.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ class TEST_API Executor {
146146
bool create_vars = true,
147147
bool keep_kids = false);
148148

149-
void EnableMKLDNN(const ProgramDesc& program);
149+
void EnableONEDNN(const ProgramDesc& program);
150150

151151
std::shared_ptr<TrainerBase> InitForDataset(
152152
const ProgramDesc& main_program,

paddle/fluid/inference/api/analysis_config.cc

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -665,7 +665,7 @@ void AnalysisConfig::EnableCUDNN() {
665665
Update();
666666
}
667667

668-
void AnalysisConfig::EnableMKLDNN() {
668+
void AnalysisConfig::EnableONEDNN() {
669669
#ifdef PADDLE_WITH_DNNL
670670
use_onednn_ = true;
671671
#else
@@ -967,7 +967,7 @@ void AnalysisConfig::Update() {
967967
}
968968

969969
#ifdef PADDLE_WITH_DNNL
970-
// Since EnableMKLDNN is default, the pass_builder has created in the first
970+
// Since EnableONEDNN is default, the pass_builder has created in the first
971971
// time.
972972
// Case1: User manually disable onednn after pass_builder
973973
// create.(config.disable_mkldnn())
@@ -1033,7 +1033,7 @@ void AnalysisConfig::Update() {
10331033
if (use_onednn_ && enable_ir_optim_) {
10341034
#ifdef PADDLE_WITH_DNNL
10351035
// default enable onednn when device is cpu and enable_ir_optim
1036-
pass_builder()->EnableMKLDNN();
1036+
pass_builder()->EnableONEDNN();
10371037
#endif
10381038
}
10391039
}

paddle/fluid/inference/api/paddle_analysis_config.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -869,7 +869,7 @@ struct PD_INFER_DECL AnalysisConfig {
869869
/// \brief Turn on OneDNN.
870870
///
871871
///
872-
void EnableMKLDNN();
872+
void EnableONEDNN();
873873

874874
///
875875
/// \brief Turn down OneDNN.

paddle/fluid/inference/api/paddle_pass_builder.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -317,7 +317,7 @@ void GpuPassStrategy::EnableCUDNN() {
317317
use_cudnn_ = true;
318318
}
319319

320-
void GpuPassStrategy::EnableMKLDNN() {
320+
void GpuPassStrategy::EnableONEDNN() {
321321
LOG(ERROR) << "GPU not support MKLDNN yet";
322322
}
323323

@@ -343,7 +343,7 @@ CpuPassStrategy::CpuPassStrategy() : PassStrategy({}) {
343343

344344
void CpuPassStrategy::EnableCUDNN() { LOG(ERROR) << "CPU not support cuDNN"; }
345345

346-
void CpuPassStrategy::EnableMKLDNN() {
346+
void CpuPassStrategy::EnableONEDNN() {
347347
// TODO(Superjomn) Consider the way to mix CPU with GPU.
348348
#ifdef PADDLE_WITH_DNNL
349349
if (!use_onednn_) {

paddle/fluid/inference/api/paddle_pass_builder.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,7 @@ class PD_INFER_DECL PassStrategy : public PaddlePassBuilder {
142142
/// \brief Enable the use of OneDNN.
143143
/// The OneDNN control exists in both CPU and GPU mode, because there can
144144
/// still be some CPU kernels running in GPU mode.
145-
virtual void EnableMKLDNN() {}
145+
virtual void EnableONEDNN() {}
146146

147147
/// \brief Disable the use of OneDNN.
148148
virtual void DisableMKLDNN() {}
@@ -211,7 +211,7 @@ class PD_INFER_DECL CpuPassStrategy : public PassStrategy {
211211
void EnableCUDNN() override;
212212

213213
/// \brief Enable the use of OneDNN.
214-
void EnableMKLDNN() override;
214+
void EnableONEDNN() override;
215215

216216
/// \brief Disable the use of OneDNN.
217217
void DisableMKLDNN() override;
@@ -257,7 +257,7 @@ class PD_INFER_DECL GpuPassStrategy : public PassStrategy {
257257
void EnableCUDNN() override;
258258

259259
/// \brief Not supported in GPU mode yet.
260-
void EnableMKLDNN() override;
260+
void EnableONEDNN() override;
261261

262262
/// \brief Not supported in GPU mode yet.
263263
void EnableMkldnnBfloat16() override;

paddle/fluid/inference/capi/paddle_c_api.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -233,7 +233,7 @@ typedef struct PD_MaxInputShape {
233233
PADDLE_CAPI_EXPORT extern void PD_SwitchIrDebug(PD_AnalysisConfig* config,
234234
bool x);
235235

236-
PADDLE_CAPI_EXPORT extern void PD_EnableMKLDNN(PD_AnalysisConfig* config);
236+
PADDLE_CAPI_EXPORT extern void PD_EnableONEDNN(PD_AnalysisConfig* config);
237237

238238
PADDLE_CAPI_EXPORT extern void PD_SetMkldnnCacheCapacity(
239239
PD_AnalysisConfig* config, int capacity);

paddle/fluid/inference/capi/pd_config.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -277,12 +277,12 @@ void PD_SwitchIrDebug(PD_AnalysisConfig* config, bool x) {
277277
config->config.SwitchIrDebug(x);
278278
}
279279

280-
void PD_EnableMKLDNN(PD_AnalysisConfig* config) {
280+
void PD_EnableONEDNN(PD_AnalysisConfig* config) {
281281
PADDLE_ENFORCE_NOT_NULL(
282282
config,
283283
common::errors::InvalidArgument(
284284
"The pointer of analysis configuration shouldn't be nullptr"));
285-
config->config.EnableMKLDNN();
285+
config->config.EnableONEDNN();
286286
}
287287

288288
void PD_SetMkldnnCacheCapacity(PD_AnalysisConfig* config, int capacity) {

paddle/fluid/inference/capi_exp/pd_config.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -363,9 +363,9 @@ void PD_ConfigSwitchIrDebug(__pd_keep PD_Config* pd_config, PD_Bool x) {
363363
CHECK_AND_CONVERT_PD_CONFIG;
364364
config->SwitchIrDebug(x);
365365
}
366-
void PD_ConfigEnableMKLDNN(__pd_keep PD_Config* pd_config) {
366+
void PD_ConfigEnableONEDNN(__pd_keep PD_Config* pd_config) {
367367
CHECK_AND_CONVERT_PD_CONFIG;
368-
config->EnableMKLDNN();
368+
config->EnableONEDNN();
369369
}
370370
void PD_ConfigSetMkldnnCacheCapacity(__pd_keep PD_Config* pd_config,
371371
int32_t capacity) {

paddle/fluid/inference/capi_exp/pd_config.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -504,7 +504,7 @@ PADDLE_CAPI_EXPORT extern void PD_ConfigSwitchIrDebug(
504504
///
505505
/// \param[in] pd_config config
506506
///
507-
PADDLE_CAPI_EXPORT extern void PD_ConfigEnableMKLDNN(
507+
PADDLE_CAPI_EXPORT extern void PD_ConfigEnableONEDNN(
508508
__pd_keep PD_Config* pd_config);
509509
///
510510
/// \brief Set the cache capacity of different input shapes for OneDNN.

0 commit comments

Comments
 (0)