fix a bug in GetTrtWeight (#48993)

zhoutianzi666 · web-flow · commit 93e36b069821 · 2022-12-12T10:28:10.000+08:00
diff --git 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(

Original file line number	Diff line number	Diff line change
`@@ -732,7 +732,7 @@ TensorRTEngine::Weight TensorRTEngine::GetTrtWeight(`
`732`	`732`	`for (int i = 0; i < weight_tensor.numel(); i++) {`
`733`	`733`	`int32_data[i] = int64_data[i];`
`734`	`734`	`}`
`735`		`- weight.SetDataType(phi::DataType::FLOAT32);`
	`735`	`+ weight.SetDataType(phi::DataType::INT32);`
`736`	`736`	`weight.SetValues(int32_data);`
`737`	`737`	`} else {`
`738`	`738`	`paddle::framework::TensorCopySync(`