fix convert weight failed. (#45346)

PaddlePaddle · Aug 24, 2022 · 3d514e4 · 3d514e4
1 parent fac8a26
commit 3d514e4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/paddle/fluid/inference/analysis/passes/convert_to_mixed_precision.cc b/paddle/fluid/inference/analysis/passes/convert_to_mixed_precision.cc
@@ -162,6 +162,8 @@ void SaveMixedModel(
     if (var->IsType<framework::LoDTensor>() ||
         var->IsType<framework::Tensor>()) {
       auto* t = var->GetMutable<framework::LoDTensor>();
+      if (t->dtype() != phi::DataType::FLOAT32) continue;
+
       framework::Tensor mixed_tensor;
       mixed_tensor.Resize(t->dims());
       auto* data = t->mutable_data<float>(platform::CPUPlace());