fix param (#50226)

708f6e79 · wenbin · GitHub · b38142bc · 708f6e79
隐藏空白更改
内联并排

Showing with 3 addition and 3 deletion

paddle/fluid/inference/tensorrt/convert/fused_lookup_tables_op.cc ...luid/inference/tensorrt/convert/fused_lookup_tables_op.cc +3 -3

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