Skip to content

Commit 1be810b

Browse files
authored
Modify kPirMkldnnPasses to kPirOnednnPasses [fluid_ops] (#74424)
1 parent c2b46ad commit 1be810b

File tree

3 files changed

+10
-10
lines changed

3 files changed

+10
-10
lines changed

paddle/fluid/inference/api/analysis_predictor.cc

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1036,19 +1036,19 @@ void AnalysisPredictor::OptimizeInferencePirProgram() {
10361036
pir::IrContext *ctx = pir::IrContext::Instance();
10371037
ctx->GetOrRegisterDialect<paddle::dialect::OneDNNOperatorDialect>();
10381038
if (!config_.custom_pass_only_) {
1039-
for (const auto &mkldnn_pass : kPirMkldnnPasses) {
1039+
for (const auto &onednn_pass : kPirOnednnPasses) {
10401040
if (std::find(config_.deleted_passes_.begin(),
10411041
config_.deleted_passes_.end(),
1042-
mkldnn_pass) == config_.deleted_passes_.end()) {
1043-
pass_pm.AddPass(pir::PassRegistry::Instance().Get(mkldnn_pass));
1042+
onednn_pass) == config_.deleted_passes_.end()) {
1043+
pass_pm.AddPass(pir::PassRegistry::Instance().Get(onednn_pass));
10441044
}
10451045
}
10461046
if (config_.onednn_bfloat16_enabled()) {
1047-
for (const auto &mkldnn_pass : kPirMkldnnBf16Passes) {
1047+
for (const auto &onednn_pass : kPirOnednnBf16Passes) {
10481048
if (std::find(config_.deleted_passes_.begin(),
10491049
config_.deleted_passes_.end(),
1050-
mkldnn_pass) == config_.deleted_passes_.end()) {
1051-
pass_pm.AddPass(pir::PassRegistry::Instance().Get(mkldnn_pass));
1050+
onednn_pass) == config_.deleted_passes_.end()) {
1051+
pass_pm.AddPass(pir::PassRegistry::Instance().Get(onednn_pass));
10521052
}
10531053
}
10541054
}

paddle/fluid/inference/api/paddle_pass_builder.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -676,7 +676,7 @@ const std::vector<std::string> kPirXpuPasses{
676676
"group_norm_silu_fuse_pass",
677677
"fc_xpu_fuse_pass"};
678678

679-
const std::vector<std::string> kPirMkldnnPasses {
679+
const std::vector<std::string> kPirOnednnPasses {
680680
"add_shadow_output_after_dead_parameter_pass",
681681
"delete_quant_dequant_linear_op_pass", //
682682
"delete_weight_dequant_linear_op_pass", //
@@ -718,7 +718,7 @@ const std::vector<std::string> kPirMkldnnPasses {
718718
"onednn_placement_pass", //
719719
};
720720

721-
const std::vector<std::string> kPirMkldnnBf16Passes{
721+
const std::vector<std::string> kPirOnednnBf16Passes{
722722
"add_shadow_output_after_dead_parameter_pass",
723723
"cpu_bfloat16_placement_pass",
724724
"cpu_bfloat16_pass",

paddle/fluid/inference/api/paddle_pass_builder.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -392,7 +392,7 @@ PD_INFER_DECL extern const std::vector<std::string> kPirCustomDevicePasses;
392392
PD_INFER_DECL extern const std::vector<std::string> kPirGpuPasses;
393393
PD_INFER_DECL extern const std::vector<std::string> kPirCpuPasses;
394394
PD_INFER_DECL extern const std::vector<std::string> kPirXpuPasses;
395-
PD_INFER_DECL extern const std::vector<std::string> kPirMkldnnPasses;
396-
PD_INFER_DECL extern const std::vector<std::string> kPirMkldnnBf16Passes;
395+
PD_INFER_DECL extern const std::vector<std::string> kPirOnednnPasses;
396+
PD_INFER_DECL extern const std::vector<std::string> kPirOnednnBf16Passes;
397397

398398
} // namespace paddle

0 commit comments

Comments
 (0)