[PaddleTRT] 支持reduce op进入trt (#57326)

vivienfanghuagood · web-flow · commit 5429e20320d0 · 2023-11-14T10:17:39.000+08:00
支持reduce op 进入trt (#57326)
diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc
@@ -2198,11 +2198,6 @@ struct SimpleOpTypeSetTeller : public Teller {
         for (auto x : dim) {
           if (x == 0 || (x + input_shape.size() == 0)) return false;
         }
-
-      } else {
-        if (PADDLE_GET_CONST(bool, desc.GetAttr("reduce_all")) &&
-            !PADDLE_GET_CONST(bool, desc.GetAttr("keep_dim")))
-          return false;
       }
 
       auto dtype = x_var_desc->GetDataType();

Original file line number	Diff line number	Diff line change
`@@ -2198,11 +2198,6 @@ struct SimpleOpTypeSetTeller : public Teller {`
`2198`	`2198`	`for (auto x : dim) {`
`2199`	`2199`	`if (x == 0 \|\| (x + input_shape.size() == 0)) return false;`
`2200`	`2200`	`}`
`2201`		`-`
`2202`		`- } else {`
`2203`		`- if (PADDLE_GET_CONST(bool, desc.GetAttr("reduce_all")) &&`
`2204`		`- !PADDLE_GET_CONST(bool, desc.GetAttr("keep_dim")))`
`2205`		`- return false;`
`2206`	`2201`	`}`
`2207`	`2202`
`2208`	`2203`	`auto dtype = x_var_desc->GetDataType();`