fix a bug in GetTrtWeight (#48993)

93e36b06 · zhoutianzi666 · GitHub · 4c0d46a8 · 93e36b06
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/inference/tensorrt/engine.cc paddle/fluid/inference/tensorrt/engine.cc +1 -1

未找到文件。
--- a/paddle/fluid/inference/tensorrt/engine.cc
+++ b/paddle/fluid/inference/tensorrt/engine.cc
@@ -732,7 +732,7 @@ TensorRTEngine::Weight TensorRTEngine::GetTrtWeight(
    for (int i = 0; i < weight_tensor.numel(); i++) {
      int32_data[i] = int64_data[i];
    }
-    weight.SetDataType(phi::DataType::FLOAT32);
+    weight.SetDataType(phi::DataType::INT32);
    weight.SetValues(int32_data);
  } else {
    paddle::framework::TensorCopySync(