Print input ID's corresponding input shape after each run (#643)

Author: yf225

tritonbench/utils/triton_op.py

@@ -1135,6 +1135,9 @@ def _reduce_benchmarks(acc, bm_name: str):
                     _reduce_benchmarks, benchmarks, {}
                 )
                 metrics.append((x_val, y_vals))
+                logger.warning(
+                    f"Completed input ID {input_id}:\n{table}",
+                )
                 self._cur_backend_name = None
                 del self.example_inputs  # save some memory
                 if "proton" in self.required_metrics: