Type lp.map_instructions.

Author: kaushikcfd

loopy/transform/instruction.py

@@ -39,7 +39,7 @@
 
 
 if TYPE_CHECKING:
-    from collections.abc import Mapping, Sequence
+    from collections.abc import Callable, Mapping, Sequence
 
     from pymbolic import ArithmeticExpression
     from pymbolic.primitives import Subscript
@@ -84,7 +84,8 @@ def find_instructions(
 
 # {{{ map_instructions
 
-def map_instructions(kernel, insn_match, f):
+def map_instructions(kernel: LoopKernel, insn_match: ToMatchConvertible,
+                     f: Callable[[InstructionBase], InstructionBase]) -> LoopKernel:
     from loopy.match import parse_match
     match = parse_match(insn_match)