Skip to content

Commit

Permalink
fix the conflicts and pytest error
Browse files Browse the repository at this point in the history
Signed-off-by: Bangtian Liu <[email protected]>
  • Loading branch information
bangtianliu committed Jan 7, 2025
1 parent 5fbde03 commit d7ff2a9
Show file tree
Hide file tree
Showing 4 changed files with 1 addition and 8 deletions.
1 change: 0 additions & 1 deletion tuner/examples/test/tuner_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,7 +91,6 @@ def main():

path_config = libtuner.PathConfig()
path_config.base_dir.mkdir(parents=True, exist_ok=True)
path_config.output_unilog.touch()
# TODO(Max191): Make candidate_trackers internal to TuningClient.
candidate_trackers: list[libtuner.CandidateTracker] = []
stop_after_phase: str = args.stop_after
Expand Down
2 changes: 0 additions & 2 deletions tuner/tuner/common.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,12 +10,10 @@
from types import TracebackType
from typing import Optional
from typing import Any
from typing_extensions import Literal

from iree.compiler import ir # type: ignore

from iree.compiler.dialects import iree_gpu # type: ignore
from iree.compiler.dialects import iree_codegen # type: ignore


class CommonTypes:
Expand Down
3 changes: 0 additions & 3 deletions tuner/tuner/dispatch_parser_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,10 +27,7 @@ def tuner_ctx() -> Generator[common.TunerContext, None, None]:
from logging import Logger
from unittest.mock import MagicMock

# Mock the logger
mock_logger = MagicMock(spec=Logger)

# Use TunerContext with the mocked logger
with common.TunerContext(logger=mock_logger) as ctx:
yield ctx

Expand Down
3 changes: 1 addition & 2 deletions tuner/tuner/libtuner.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,15 +18,13 @@

import math
import signal
import subprocess
import sys
import shutil
import logging
import argparse
from datetime import datetime
from enum import Enum
from pathlib import Path
import time
import multiprocessing
import queue
from tqdm import tqdm
Expand All @@ -37,6 +35,7 @@
import iree.runtime as ireert # type: ignore
import iree.compiler as ireec # type: ignore
from iree.compiler import ir # type: ignore
from iree.compiler.dialects import iree_codegen # type: ignore
from . import candidate_gen
from . import dispatch_parser
from .op_matchers import *
Expand Down

0 comments on commit d7ff2a9

Please sign in to comment.