未验证 提交 77c9c90a 编写于 作者: Z Zhang Jun 提交者: GitHub

[inference][trt]reduce support int64 (#50897)

上级 8a6de610
......@@ -2213,14 +2213,8 @@ struct SimpleOpTypeSetTeller : public Teller {
return false;
}
#if IS_TRT_VERSION_LT(7000)
auto dtype = x_var_desc->GetDataType();
#if IS_TRT_VERSION_GE(7000)
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;
}
#else
if (dtype != framework::proto::VarType::FP32) {
VLOG(3) << "reduce op input data type must be float32 using TensorRT "
"< 7.0";
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册