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 80e89185c2d2fa96e028bec16ab57543506b5b5b..c8676b6c191e45779dee85ad0f2ae40ee3f6dadd 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(w_dims[1]);