From fec3c5b97a551c66f471ea481e495e5186b19568 Mon Sep 17 00:00:00 2001 From: zhangjun Date: Thu, 20 Jan 2022 15:45:23 +0000 Subject: [PATCH] update --- paddle/fluid/inference/tensorrt/op_teller.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc index ffd2f579b6429..fcb4890690524 100644 --- a/paddle/fluid/inference/tensorrt/op_teller.cc +++ b/paddle/fluid/inference/tensorrt/op_teller.cc @@ -1519,8 +1519,8 @@ bool OpTeller::Tell(const framework::ir::Node* node, bool use_no_calib_int8, } auto dtype = x_var_desc->GetDataType(); - if (dtype != - framework::proto::VarType::INT32 && dtype != framework::proto::VarType::FP32) { + if (dtype != framework::proto::VarType::INT32 && + dtype != framework::proto::VarType::FP32) { VLOG(3) << "reduce op input data type must be int32 or float32"; return false; }