未验证 提交 708f6e79 编写于 作者: W wenbin 提交者: GitHub

fix param (#50226)

上级 b38142bc
...@@ -70,11 +70,11 @@ class FusedLookupTablesOpConverter : public OpConverter { ...@@ -70,11 +70,11 @@ class FusedLookupTablesOpConverter : public OpConverter {
auto w_dims = w_tensor->dims(); auto w_dims = w_tensor->dims();
weight = engine_->GetTrtWeight(w_name, *w_tensor); weight = engine_->GetTrtWeight(w_name, *w_tensor);
auto weight_size = phi::product(w_dims); auto weight_size = phi::product(w_dims);
bool output_fp16; int output_fp16;
if (engine_->precision() == AnalysisConfig::Precision::kFloat32) { if (engine_->precision() == AnalysisConfig::Precision::kFloat32) {
output_fp16 = false; output_fp16 = 0;
} else { } else {
output_fp16 = true; output_fp16 = 1;
} }
int32_t weight_width = static_cast<int32_t>(w_dims[1]); int32_t weight_width = static_cast<int32_t>(w_dims[1]);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册