[Paddle-TRT] Fix cast (#44312)

* fix_cast * fix_cast * commit

[Paddle-TRT] Fix cast (#44312)
* fix_cast * fix_cast * commit
7a85ced3 · zhoutianzi666 · GitHub · 5c291737 · 7a85ced3
显示空白变更内容
内联并排

Showing with 4 addition and 0 deletion

paddle/fluid/inference/tensorrt/op_teller.cc paddle/fluid/inference/tensorrt/op_teller.cc +4 -0

未找到文件。
--- 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) {