Skip to content

Commit e29a10c

Browse files
authored
fix issue of no op_info when enabling oneDNN (#71426)
1 parent 6a1615c commit e29a10c

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

paddle/fluid/inference/api/analysis_predictor.cc

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -109,6 +109,10 @@
109109
#include "paddle/pir/include/dialect/shape/utils/shape_analysis.h"
110110
#endif
111111

112+
#ifdef PADDLE_WITH_DNNL
113+
#include "paddle/fluid/pir/dialect/operator/ir/op_onednn_dialect.h"
114+
#endif
115+
112116
#include "paddle/common/flags.h"
113117
#include "paddle/fluid/ir_adaptor/translator/translate.h"
114118
#include "paddle/fluid/pir/dialect/operator/ir/pd_op.h"
@@ -1008,6 +1012,8 @@ void AnalysisPredictor::OptimizeInferencePirProgram() {
10081012
#ifdef PADDLE_WITH_DNNL
10091013
} else if (config_.mkldnn_enabled()) {
10101014
// mkldnn
1015+
pir::IrContext *ctx = pir::IrContext::Instance();
1016+
ctx->GetOrRegisterDialect<paddle::dialect::OneDNNOperatorDialect>();
10111017
if (!config_.custom_pass_only_) {
10121018
for (const auto &mkldnn_pass : kPirMkldnnPasses) {
10131019
if (std::find(config_.deleted_passes_.begin(),

0 commit comments

Comments
 (0)