Remove unused imports

Signed-off-by: Jakub Kuderski <[email protected]>
nod-ai · Nov 14, 2024 · fed787a · fed787a
1 parent b800d1c
commit fed787a
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/tuner/tuner/candidate_gen.py b/tuner/tuner/candidate_gen.py
@@ -32,7 +32,6 @@
 from abc import ABC, abstractmethod
 
 from iree.compiler import ir  # type: ignore
-from iree.compiler.dialects import iree_codegen, iree_gpu  # type: ignore
 
 tune_logger = logging.getLogger("tune")
 

diff --git a/tuner/tuner/candidate_gen_test.py b/tuner/tuner/candidate_gen_test.py
@@ -12,7 +12,7 @@
 from . import candidate_gen
 
 from iree.compiler import ir  # type: ignore
-from iree.compiler.dialects import func, iree_codegen, iree_gpu  # type: ignore
+from iree.compiler.dialects import func # type: ignore
 
 
 def test_get_shaped_type_element_bitwidth() -> None: