Skip to content

Commit

Permalink
Remove unet strings
Browse files Browse the repository at this point in the history
  • Loading branch information
RattataKing committed Aug 19, 2024
1 parent 109f39f commit 21f4976
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions tuning/autotune.py
Original file line number Diff line number Diff line change
Expand Up @@ -86,7 +86,7 @@ class PathConfig:
# Preset constants
global_config_prolog_mlir: Path = Path("./config_prolog.mlir")
global_config_epilog_mlir: Path = Path("./config_epilog.mlir")
model_baseline_vmfb: Path = Path("./unet_baseline.vmfb")
model_baseline_vmfb: Path = Path("./baseline.vmfb")

# Dynamic paths
base_dir: Path = field(init=False)
Expand Down Expand Up @@ -260,7 +260,7 @@ def generate_sample_result(


@dataclass
class UnetBenchmarkResult:
class ModelBenchmarkResult:
result_str: Optional[str] = None

def get_tokens(self) -> list[str]:
Expand Down Expand Up @@ -1155,7 +1155,7 @@ def parse_grouped_benchmark_results(
if result_str is None:
continue

res = UnetBenchmarkResult(result_str)
res = ModelBenchmarkResult(result_str)
device_id = res.get_device_id()

# Record baseline benchmarking result.
Expand Down Expand Up @@ -1224,7 +1224,7 @@ def generate_dryrun_unet_benchmark_results(
task_result = subprocess.CompletedProcess(
args=[""],
returncode=0,
stdout=UnetBenchmarkResult().generate_sample_result(
stdout=ModelBenchmarkResult().generate_sample_result(
candidate_vmfb_path_str=candidate_vmfb_path.as_posix(),
device_id=device_id,
t1=start,
Expand Down

0 comments on commit 21f4976

Please sign in to comment.