feat(typing): allow expressions as predicates in instructions

Author: alexfikl

loopy/kernel/instruction.py

@@ -304,7 +304,7 @@ def __init__(self,
                  within_inames_is_final: bool | None,
                  within_inames: frozenset[str] | None,
                  priority: int | None,
-                 predicates: frozenset[str] | None,
+                 predicates: frozenset[str] | frozenset[Expression] | None,
                  tags: frozenset[Tag] | None,
                  *,
                  depends_on: frozenset[str] | str | None = None,
@@ -946,7 +946,7 @@ def __init__(self,
                  within_inames_is_final: bool | None = None,
                  within_inames: frozenset[str] | None = None,
                  priority: int | None = None,
-                 predicates: frozenset[str] | None = None,
+                 predicates: frozenset[str] | frozenset[Expression] | None = None,
                  tags: frozenset[Tag] | None = None,
                  temp_var_type:
                     type[_not_provided]