diff --git a/torch/_inductor/choices.py b/torch/_inductor/choices.py index 05ad3beb9aa..d8a05d88c06 100644 --- a/torch/_inductor/choices.py +++ b/torch/_inductor/choices.py @@ -24,8 +24,6 @@ from .virtualized import V if TYPE_CHECKING: - from functools import partial - from triton import Config as TritonConfig from torch.utils._ordered_set import OrderedSet