fix convert weight failed. (#45346)

3d514e48 · Wilber · GitHub · fac8a260 · 3d514e48
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

paddle/fluid/inference/analysis/passes/convert_to_mixed_precision.cc ...d/inference/analysis/passes/convert_to_mixed_precision.cc +2 -0

未找到文件。
--- 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());