Skip to content

Commit 203754e

Browse files
authoredDec 26, 2023
tile (#60261)
1 parent 20d3558 commit 203754e

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed
 

‎paddle/fluid/inference/tensorrt/op_teller.cc

100755100644
+7-1
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,7 @@ struct SimpleOpTypeSetTeller : public Teller {
4747
#endif
4848
#if IS_TRT_VERSION_GE(7000)
4949
teller_set.insert("tile");
50+
int8_teller_set.insert("tile");
5051
teller_set.insert("flatten_contiguous_range");
5152
int8_teller_set.insert("flatten_contiguous_range");
5253
teller_set.insert("rnn");
@@ -2302,15 +2303,20 @@ struct SimpleOpTypeSetTeller : public Teller {
23022303
if (!with_dynamic_shape) {
23032304
if (tile_inputs.find("repeat_times_tensor") != tile_inputs.end()) {
23042305
if (!desc.Input("repeat_times_tensor").empty()) {
2306+
VLOG(3) << "Tile op: repeat_times_tensor is not empty.";
23052307
return false;
23062308
}
23072309
}
23082310
if (tile_inputs.find("RepeatTimes") != tile_inputs.end()) {
23092311
if (!desc.Input("RepeatTimes").empty()) {
2312+
VLOG(3) << "Tile op: RepeatTimes is not empty.";
23102313
return false;
23112314
}
23122315
}
2313-
if (!desc.HasAttr("repeat_times")) return false;
2316+
if (!desc.HasAttr("repeat_times")) {
2317+
VLOG(3) << "Tile op:`repeat_times` is not set.";
2318+
return false;
2319+
}
23142320
}
23152321
}
23162322
#endif

0 commit comments

Comments
 (0)
Please sign in to comment.