Track {set,fset}_union move to pytools

Author: inducer

loopy/check.py

@@ -35,7 +35,7 @@
 import islpy as isl
 from islpy import dim_type
 from pymbolic.primitives import AlgebraicLeaf, Variable, is_arithmetic_expression
-from pytools import memoize_method
+from pytools import memoize_method, set_union
 
 from loopy.diagnostic import (
     LoopyError,
@@ -72,7 +72,7 @@
     check_each_kernel,
 )
 from loopy.type_inference import TypeReader
-from loopy.typing import auto, not_none, set_union
+from loopy.typing import auto, not_none
 
 
 if TYPE_CHECKING:

loopy/kernel/__init__.py

@@ -53,6 +53,7 @@
 from islpy import dim_type
 from pytools import (
     UniqueNameGenerator,
+    fset_union,
     generate_unique_names,
     memoize_method,
     natsorted,
@@ -72,7 +73,7 @@
 )
 from loopy.tools import update_persistent_hash
 from loopy.types import LoopyType, NumpyType
-from loopy.typing import InsnId, PreambleGenerator, SymbolMangler, fset_union, not_none
+from loopy.typing import InsnId, PreambleGenerator, SymbolMangler, not_none
 
 
 if TYPE_CHECKING:

loopy/kernel/tools.py

@@ -47,7 +47,7 @@
 import pymbolic.primitives as p
 from islpy import dim_type
 from pymbolic import Expression
-from pytools import memoize_on_first_arg, natsorted
+from pytools import fset_union, memoize_on_first_arg, natsorted, set_union
 
 from loopy.diagnostic import LoopyError, warn_with_kernel
 from loopy.kernel import LoopKernel
@@ -66,7 +66,6 @@
     TUnitOrKernelT,
     for_each_kernel,
 )
-from loopy.typing import fset_union, set_union
 
 
 if TYPE_CHECKING:

loopy/schedule/tools.py

@@ -70,12 +70,12 @@
 from constantdict import constantdict
 
 import islpy as isl
-from pytools import memoize_method, memoize_on_first_arg
+from pytools import fset_union, memoize_method, memoize_on_first_arg
 
 from loopy.diagnostic import LoopyError
 from loopy.kernel.data import AddressSpace, ArrayArg, TemporaryVariable
 from loopy.schedule.tree import Tree
-from loopy.typing import InameStr, InameStrSet, fset_union, not_none
+from loopy.typing import InameStr, InameStrSet, not_none
 
 
 if TYPE_CHECKING:

loopy/transform/loop_fusion.py

@@ -32,7 +32,7 @@
 from constantdict import constantdict
 from typing_extensions import override
 
-from pytools import memoize_on_first_arg
+from pytools import fset_union, memoize_on_first_arg
 
 from loopy.diagnostic import LoopyError
 from loopy.kernel import LoopKernel
@@ -42,7 +42,6 @@
     RuleAwareIdentityMapper,
     SubstitutionRuleMappingContext,
 )
-from loopy.typing import InsnId, fset_union
 
 
 if TYPE_CHECKING:
@@ -51,7 +50,7 @@
     from loopy.kernel.instruction import InstructionBase
     from loopy.match import RuleStack
     from loopy.schedule.tools import LoopNestTree
-    from loopy.typing import InameStr, InameStrSet
+    from loopy.typing import InameStr, InameStrSet, InsnId
 
 
 __doc__ = """

loopy/typing.py

@@ -50,17 +50,16 @@
 """
 
 
-from typing import TYPE_CHECKING, TypeAlias, TypeVar, cast
+from typing import TYPE_CHECKING, TypeAlias, TypeVar
 
 import numpy as np
-from typing_extensions import TypeIs
-
 from pymbolic.primitives import ExpressionNode
 from pymbolic.typing import ArithmeticExpression, Expression, Integer
+from typing_extensions import TypeIs
 
 
 if TYPE_CHECKING:
-    from collections.abc import Callable, Iterable, Iterator
+    from collections.abc import Callable, Iterator
 
     from loopy.codegen import PreambleInfo
     from loopy.kernel import LoopKernel
@@ -124,11 +123,3 @@ def integer_expr_or_err(expr: Expression) -> Integer | ExpressionNode:
 def assert_tuple(obj: tuple[ElT, ...] | object) -> tuple[ElT, ...]:
     assert isinstance(obj, tuple)
     return obj
-
-
-def set_union(iterable: Iterable[Iterable[T]]):
-    return cast("set[T]", set()).union(*iterable)
-
-
-def fset_union(iterable: Iterable[Iterable[T]]):
-    return cast("frozenset[T]", frozenset()).union(*iterable)

pyproject.toml

@@ -28,7 +28,7 @@ classifiers = [
     "Topic :: Utilities",
 ]
 dependencies = [
-    "pytools>=2025.1.6",
+    "pytools>=2025.2.2",
     "pymbolic>=2024.2.2",
     "genpy>=2016.1.2",