tile (PaddlePaddle#60261)

hanhaowen-mt · Dec 26, 2023 · 203754e · 203754e
1 parent 20d3558
commit 203754e
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc
@@ -47,6 +47,7 @@ struct SimpleOpTypeSetTeller : public Teller {
 #endif
 #if IS_TRT_VERSION_GE(7000)
     teller_set.insert("tile");
+    int8_teller_set.insert("tile");
     teller_set.insert("flatten_contiguous_range");
     int8_teller_set.insert("flatten_contiguous_range");
     teller_set.insert("rnn");
@@ -2302,15 +2303,20 @@ struct SimpleOpTypeSetTeller : public Teller {
       if (!with_dynamic_shape) {
         if (tile_inputs.find("repeat_times_tensor") != tile_inputs.end()) {
           if (!desc.Input("repeat_times_tensor").empty()) {
+            VLOG(3) << "Tile op: repeat_times_tensor is not empty.";
             return false;
           }
         }
         if (tile_inputs.find("RepeatTimes") != tile_inputs.end()) {
           if (!desc.Input("RepeatTimes").empty()) {
+            VLOG(3) << "Tile op: RepeatTimes is not empty.";
             return false;
           }
         }
-        if (!desc.HasAttr("repeat_times")) return false;
+        if (!desc.HasAttr("repeat_times")) {
+          VLOG(3) << "Tile op:`repeat_times` is not set.";
+          return false;
+        }
       }
     }
 #endif