Skip to content

Commit

Permalink
Remove extra dataclass labels
Browse files Browse the repository at this point in the history
  • Loading branch information
RattataKing committed Aug 19, 2024
1 parent d420c46 commit cd5ae2a
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 4 deletions.
1 change: 0 additions & 1 deletion tuning/autotune.py
Original file line number Diff line number Diff line change
Expand Up @@ -157,7 +157,6 @@ def get_compiled_model_index(self, file_path: Path) -> int:
return int(file_path.stem.split("_")[-1])


@dataclass
class TuningClient(ABC):
@abstractmethod
def get_dispatch_compile_command(
Expand Down
4 changes: 1 addition & 3 deletions tuning/punet_autotune.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,11 +5,9 @@
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

import autotune
from dataclasses import dataclass
from pathlib import Path


@dataclass
class PunetClient(autotune.TuningClient):

def get_dispatch_compile_command(
Expand Down Expand Up @@ -97,7 +95,7 @@ def main():
args, path_config, compiled_candidates, candidate_trackers, punet_client
)
print(f"Stored results in {path_config.output_unilog}\n")
if stop_after_phase == ExecutionPhases.benchmark_dispatches:
if stop_after_phase == autotune.ExecutionPhases.benchmark_dispatches:
return

print(f"Compiling top model candidates...")
Expand Down

0 comments on commit cd5ae2a

Please sign in to comment.