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