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

correct define (#33966)

* correct op

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