Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
zhangjun committed Jan 20, 2022
1 parent c030eb7 commit fec3c5b
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions paddle/fluid/inference/tensorrt/op_teller.cc
Original file line number Diff line number Diff line change
Expand Up @@ -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;
}
Expand Down

0 comments on commit fec3c5b

Please sign in to comment.