From 708f6e7968fd8cc474be16708658a413331d4433 Mon Sep 17 00:00:00 2001
From: wenbin <wang3323032@qq.com>
Date: Mon, 6 Feb 2023 19:07:59 +0800
Subject: [PATCH] fix param (#50226)

---
 .../inference/tensorrt/convert/fused_lookup_tables_op.cc    | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/paddle/fluid/inference/tensorrt/convert/fused_lookup_tables_op.cc b/paddle/fluid/inference/tensorrt/convert/fused_lookup_tables_op.cc
index 80e89185c2..c8676b6c19 100644
--- a/paddle/fluid/inference/tensorrt/convert/fused_lookup_tables_op.cc
+++ b/paddle/fluid/inference/tensorrt/convert/fused_lookup_tables_op.cc
@@ -70,11 +70,11 @@ class FusedLookupTablesOpConverter : public OpConverter {
     auto w_dims = w_tensor->dims();
     weight = engine_->GetTrtWeight(w_name, *w_tensor);
     auto weight_size = phi::product(w_dims);
-    bool output_fp16;
+    int output_fp16;
     if (engine_->precision() == AnalysisConfig::Precision::kFloat32) {
-      output_fp16 = false;
+      output_fp16 = 0;
     } else {
-      output_fp16 = true;
+      output_fp16 = 1;
     }
 
     int32_t weight_width = static_cast<int32_t>(w_dims[1]);
-- 
GitLab