Skip to content

Commit d04c203

Browse files
committed
Let user know which autotune_effort is being used.
1 parent 09ccf1a commit d04c203

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

helion/autotuner/base_cache.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -178,7 +178,10 @@ def autotune(self, *, skip_cache: bool = False) -> Config:
178178
counters["autotune"]["cache_miss"] += 1
179179
log.debug("cache miss")
180180

181-
self.autotuner.log("Starting autotuning process, this may take a while...")
181+
effort = self.kernel.settings.autotune_effort
182+
self.autotuner.log(
183+
f"Starting autotuning process with effort={effort}, this may take a while..."
184+
)
182185

183186
config = self.autotuner.autotune()
184187

helion/runtime/kernel.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -578,7 +578,7 @@ def _implicit_config(self) -> Config | None:
578578
if not is_ref_mode_enabled(self.kernel.settings):
579579
kernel_decorator = self.format_kernel_decorator(config, self.settings)
580580
print(
581-
f"Using default config: {kernel_decorator}",
581+
f"Using default config (autotune_effort=none): {kernel_decorator}",
582582
file=sys.stderr,
583583
)
584584
return config

0 commit comments

Comments
 (0)