Skip to content

Commit 75f5f6e

Browse files
committed
fix bug
1 parent 33151ba commit 75f5f6e

File tree

180 files changed

+6948886
-17
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

180 files changed

+6948886
-17
lines changed

benchmarks/run.py

Lines changed: 0 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -316,23 +316,6 @@ def run_kernel_variants(
316316
) -> None:
317317
"""Run kernel variants in the same benchmark run."""
318318

319-
# Configure Helion to use fewer generations for faster autotuning during benchmarks
320-
import helion
321-
from helion.autotuner import DifferentialEvolutionSearch, LocalAutotuneCache
322-
from helion.runtime.kernel import BoundKernel
323-
from typing import Sequence
324-
325-
def fast_autotuner_fn(
326-
bound_kernel: BoundKernel, args: Sequence[object], **kwargs: object
327-
) -> LocalAutotuneCache:
328-
# Use only 1 generation instead of default 20 for faster benchmarking
329-
return LocalAutotuneCache(
330-
DifferentialEvolutionSearch(bound_kernel, args, num_generations=1, **kwargs)
331-
)
332-
333-
# Set the custom autotuner function
334-
helion.set_default_settings(helion.Settings(autotuner_fn=fast_autotuner_fn))
335-
336319
# Import tritonbench components
337320
try:
338321
from tritonbench.utils.parser import ( # pyright: ignore[reportMissingImports]

benchmarks_autotune.txt

Lines changed: 378 additions & 0 deletions
Large diffs are not rendered by default.

benchmarks_autotune2.txt

Lines changed: 2441 additions & 0 deletions
Large diffs are not rendered by default.

benchmarks_autotune3_0.txt

Lines changed: 317 additions & 0 deletions
Large diffs are not rendered by default.

benchmarks_autotune3_1.txt

Lines changed: 254 additions & 0 deletions
Large diffs are not rendered by default.

benchmarks_autotune3_2.txt

Lines changed: 75527 additions & 0 deletions
Large diffs are not rendered by default.

benchmarks_autotune3_3.txt

Lines changed: 14465 additions & 0 deletions
Large diffs are not rendered by default.

benchmarks_autotune_10.txt

Lines changed: 1613 additions & 0 deletions
Large diffs are not rendered by default.

benchmarks_autotune_11.txt

Lines changed: 908896 additions & 0 deletions
Large diffs are not rendered by default.

benchmarks_autotune_12_notune.txt

Lines changed: 2674 additions & 0 deletions
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)