diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc index 59b196e3d92be5d4038c6419b0a689c769004b3f..c21ef8840de1299adda62dda10ddb2b92f824bfa 100644 --- a/paddle/fluid/inference/tensorrt/op_teller.cc +++ b/paddle/fluid/inference/tensorrt/op_teller.cc @@ -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; }