From 19feba386c0ddb5eaf16c1e75dc264a347d55a1b Mon Sep 17 00:00:00 2001
From: Chen Weihang <chenweihang@baidu.com>
Date: Wed, 26 Oct 2022 21:51:49 -0500
Subject: [PATCH] Fix compile error of mkldnn and tensorrt (#47388)

* fix compile error of mkldnn

* fix tensorrt error
---
 .../framework/ir/mkldnn/operator_scale_onednn_fuse_pass.cc    | 2 +-
 .../tensorrt/convert/preln_layernorm_shift_partition_op.cc    | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/paddle/fluid/framework/ir/mkldnn/operator_scale_onednn_fuse_pass.cc b/paddle/fluid/framework/ir/mkldnn/operator_scale_onednn_fuse_pass.cc
index 2759c79b7a..2e4163feb8 100644
--- a/paddle/fluid/framework/ir/mkldnn/operator_scale_onednn_fuse_pass.cc
+++ b/paddle/fluid/framework/ir/mkldnn/operator_scale_onednn_fuse_pass.cc
@@ -73,7 +73,7 @@ void FuseOperatorScaleOneDNNPass::FuseScale(Graph *graph,
       auto *scale_var = scope->FindVar(scale_var_name);
       // ScaleTensor must be weight
       if (scale_var == nullptr) return;
-      auto *scale_tensor = scale_var->GetMutable<LoDTensor>();
+      auto *scale_tensor = scale_var->GetMutable<phi::DenseTensor>();
       scale = *(scale_tensor->data<float>());
     }
 
diff --git a/paddle/fluid/inference/tensorrt/convert/preln_layernorm_shift_partition_op.cc b/paddle/fluid/inference/tensorrt/convert/preln_layernorm_shift_partition_op.cc
index 7b7719e3ac..402cd363ab 100644
--- a/paddle/fluid/inference/tensorrt/convert/preln_layernorm_shift_partition_op.cc
+++ b/paddle/fluid/inference/tensorrt/convert/preln_layernorm_shift_partition_op.cc
@@ -49,8 +49,8 @@ class PrelnLayerNormShiftPartitionOpConverter : public OpConverter {
             ? PADDLE_GET_CONST(int, op_desc.GetAttr("shift_size"))
             : 0;
 
-    auto* Bias_t = Bias_v->GetMutable<framework::LoDTensor>();
-    auto* Scale_t = Scale_v->GetMutable<framework::LoDTensor>();
+    auto* Bias_t = Bias_v->GetMutable<phi::DenseTensor>();
+    auto* Scale_t = Scale_v->GetMutable<phi::DenseTensor>();
 
     auto bias_weight =
         engine_->GetFp32TrtWeight(op_desc.Input("Bias").front(), *Bias_t);
-- 
GitLab