fix a bug in GetTrtWeight (PaddlePaddle#48993)

lxsbupt · Dec 12, 2022 · e887f93 · e887f93
1 parent 3bdfe12
commit e887f93
Showing 1 changed file with 1 addition and 1 deletion.
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(