Skip to content

Commit

Permalink
Add log.info and remove unneeded import
Browse files Browse the repository at this point in the history
  • Loading branch information
RattataKing committed Aug 7, 2024
1 parent b1ff9be commit 6c733da
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tuning/autotune.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,10 +18,8 @@
from dataclasses import dataclass, field
from typing import Type, Optional, Callable, Iterable, Any
import pickle
from itertools import groupby
import iree.runtime as ireert
import random
import tempfile

"""
Sample Usage:
Expand Down Expand Up @@ -905,6 +903,7 @@ def benchmark_compiled_candidates(
logging.info("benchmark_top_candidates()")

if args.dry_run:
logging.info("generate_dryrun_dispatch_benchmark_results")
benchmark_results = generate_dryrun_dispatch_benchmark_results(
compiled_candidates
)
Expand Down Expand Up @@ -1150,6 +1149,7 @@ def parse_grouped_benchmark_results(
def generate_dryrun_unet_benchmark_results(
unet_vmfb_paths: list[Path],
) -> list[TaskResult]:
logging.info("generate_dryrun_unet_benchmark_results")
task_results = []
start = random.uniform(100.0, 500.0)
device_id = 0
Expand Down

0 comments on commit 6c733da

Please sign in to comment.