Skip to content

Commit eae44a8

Browse files
committed
fix
1 parent 6cc79e0 commit eae44a8

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

lightllm/common/triton_utils/autotuner.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -167,7 +167,7 @@ def _autotune(self, args, kwargs, static_key, run_key):
167167

168168
bar = tqdm(
169169
self.configs,
170-
desc=f"Autotuning {self.name} for {run_key}",
170+
desc=f"Autotuning {self.kernel_name} for {run_key}",
171171
position=get_global_rank(),
172172
dynamic_ncols=True,
173173
)
@@ -179,7 +179,9 @@ def _autotune(self, args, kwargs, static_key, run_key):
179179
if run_time < best_time:
180180
best_time = run_time
181181
_best_config = config
182-
bar.set_description(f"Autotuning {self.name} [rank:{rank_id}] for {run_key}, best_time: {best_time:.5f}")
182+
bar.set_description(
183+
f"Autotuning {self.kernel_name} [rank:{rank_id}] for {run_key}, best_time: {best_time:.5f}"
184+
)
183185

184186
world_size = dist.get_world_size() if dist.is_initialized() else 1
185187
if world_size > 1:

0 commit comments

Comments
 (0)