We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e5ce286 commit f992d15Copy full SHA for f992d15
lightllm/common/triton_utils/autotuner.py
@@ -232,8 +232,6 @@ def _benchmark(_run_key):
232
kwargs_with_config["run_config"] = config
233
run_time = self._bench(*args, current_best_ms=best_time, **kwargs_with_config)
234
if run_time < best_time:
235
- if rank0 and best_time != float("inf"):
236
- logger.info(f"Best config for {self.name} is {_best_config} with time {best_time:.5f} -> {run_time:.5f}")
237
best_time = run_time
238
_best_config = config
239
bar.set_description(f"Autotuning {self.name} [rank:{rank_id}] for {_run_key}, es:{self.early_stop_cnt / len(self.configs):.2%}, best_time: {best_time:.5f}")
0 commit comments