Skip to content

Commit 31c11e0

Browse files
committed
Change autotuner logging
Signed-off-by: Amir Klein <[email protected]>
1 parent 6462feb commit 31c11e0

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

flashinfer/autotuner.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -482,8 +482,12 @@ def choose_one(
482482
)
483483
except Exception as e:
484484
shapes = self._get_input_sizes(tensors)
485+
logger.warning(
486+
f"[Autotuner]: Skipping tactic {r} {tac}, due to failure while profiling."
487+
)
485488

486-
logger.error(
489+
# Log stacktrace as debug to not spam log
490+
logger.debug(
487491
f"[Autotuner]: Failed when profiling {r} {tac}, shapes={shapes}. Error occurred: {e}"
488492
)
489493

0 commit comments

Comments
 (0)