File tree Expand file tree Collapse file tree 3 files changed +10
-10
lines changed
paddle/fluid/inference/api Expand file tree Collapse file tree 3 files changed +10
-10
lines changed Original file line number Diff line number Diff line change @@ -1036,19 +1036,19 @@ void AnalysisPredictor::OptimizeInferencePirProgram() {
1036
1036
pir::IrContext *ctx = pir::IrContext::Instance ();
1037
1037
ctx->GetOrRegisterDialect <paddle::dialect::OneDNNOperatorDialect>();
1038
1038
if (!config_.custom_pass_only_ ) {
1039
- for (const auto &mkldnn_pass : kPirMkldnnPasses ) {
1039
+ for (const auto &onednn_pass : kPirOnednnPasses ) {
1040
1040
if (std::find (config_.deleted_passes_ .begin (),
1041
1041
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 ));
1044
1044
}
1045
1045
}
1046
1046
if (config_.onednn_bfloat16_enabled ()) {
1047
- for (const auto &mkldnn_pass : kPirMkldnnBf16Passes ) {
1047
+ for (const auto &onednn_pass : kPirOnednnBf16Passes ) {
1048
1048
if (std::find (config_.deleted_passes_ .begin (),
1049
1049
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 ));
1052
1052
}
1053
1053
}
1054
1054
}
Original file line number Diff line number Diff line change @@ -676,7 +676,7 @@ const std::vector<std::string> kPirXpuPasses{
676
676
" group_norm_silu_fuse_pass" ,
677
677
" fc_xpu_fuse_pass" };
678
678
679
- const std::vector<std::string> kPirMkldnnPasses {
679
+ const std::vector<std::string> kPirOnednnPasses {
680
680
" add_shadow_output_after_dead_parameter_pass" ,
681
681
" delete_quant_dequant_linear_op_pass" , //
682
682
" delete_weight_dequant_linear_op_pass" , //
@@ -718,7 +718,7 @@ const std::vector<std::string> kPirMkldnnPasses {
718
718
" onednn_placement_pass" , //
719
719
};
720
720
721
- const std::vector<std::string> kPirMkldnnBf16Passes {
721
+ const std::vector<std::string> kPirOnednnBf16Passes {
722
722
" add_shadow_output_after_dead_parameter_pass" ,
723
723
" cpu_bfloat16_placement_pass" ,
724
724
" cpu_bfloat16_pass" ,
Original file line number Diff line number Diff line change @@ -392,7 +392,7 @@ PD_INFER_DECL extern const std::vector<std::string> kPirCustomDevicePasses;
392
392
PD_INFER_DECL extern const std::vector<std::string> kPirGpuPasses ;
393
393
PD_INFER_DECL extern const std::vector<std::string> kPirCpuPasses ;
394
394
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 ;
397
397
398
398
} // namespace paddle
You can’t perform that action at this time.
0 commit comments