未验证 提交 70ecf3b1 编写于 作者: W wenbin 提交者: GitHub

correct define (#33966)

* correct op

* test=allcase
上级 72af57bb
......@@ -51,7 +51,7 @@ struct SimpleOpTypeSetTeller : public Teller {
#if IS_TRT_VERSION_GE(7130)
teller_set.insert("group_norm");
#endif
#if CUDA_VERSION >= 10200
#if CUDA_VERSION >= 10020
teller_set.insert("reshape");
teller_set.insert("reshape2");
#endif
......@@ -719,6 +719,8 @@ bool OpTeller::Tell(const framework::ir::Node* node, bool use_no_calib_int8,
if ((*teller)(op_type, desc, use_no_calib_int8)) return true;
}
VLOG(3) << "trt unsupported op " << op_type;
return false;
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册