File tree Expand file tree Collapse file tree 1 file changed +6
-0
lines changed
paddle/fluid/inference/api Expand file tree Collapse file tree 1 file changed +6
-0
lines changed Original file line number Diff line number Diff line change 109
109
#include " paddle/pir/include/dialect/shape/utils/shape_analysis.h"
110
110
#endif
111
111
112
+ #ifdef PADDLE_WITH_DNNL
113
+ #include " paddle/fluid/pir/dialect/operator/ir/op_onednn_dialect.h"
114
+ #endif
115
+
112
116
#include " paddle/common/flags.h"
113
117
#include " paddle/fluid/ir_adaptor/translator/translate.h"
114
118
#include " paddle/fluid/pir/dialect/operator/ir/pd_op.h"
@@ -1008,6 +1012,8 @@ void AnalysisPredictor::OptimizeInferencePirProgram() {
1008
1012
#ifdef PADDLE_WITH_DNNL
1009
1013
} else if (config_.mkldnn_enabled ()) {
1010
1014
// mkldnn
1015
+ pir::IrContext *ctx = pir::IrContext::Instance ();
1016
+ ctx->GetOrRegisterDialect <paddle::dialect::OneDNNOperatorDialect>();
1011
1017
if (!config_.custom_pass_only_ ) {
1012
1018
for (const auto &mkldnn_pass : kPirMkldnnPasses ) {
1013
1019
if (std::find (config_.deleted_passes_ .begin (),
You can’t perform that action at this time.
0 commit comments