diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc index 05d0b41f14e1f4cd39d17138ca7280f0b06da250..894e44fda9496786f72cfae2123e679930b5abf0 100644 --- a/paddle/fluid/inference/tensorrt/op_teller.cc +++ b/paddle/fluid/inference/tensorrt/op_teller.cc @@ -2061,6 +2061,10 @@ bool OpTeller::Tell(const framework::ir::Node* node, } if (op_type == "cast") { +// trt 6015 result in Windows ppyolo_mbv3 TRT fp32 diff +#if !IS_TRT_VERSION_GE(7000) + return false; +#endif int in_dtype = BOOST_GET_CONST(int, desc.GetAttr("in_dtype")); int out_dtype = BOOST_GET_CONST(int, desc.GetAttr("out_dtype")); if ((in_dtype == 4 || in_dtype == 5) && out_dtype == 4) {