This commit is contained in:
Jack Taylor 2025-02-07 13:19:10 +00:00 committed by PyTorch MergeBot
parent 12a19c25b7
commit c11ed0ac59

View file

@ -1,7 +1,6 @@
from __future__ import annotations
import typing
from functools import partial
from typing import Any, Generator, Optional, TYPE_CHECKING
import sympy
@ -24,6 +23,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