InstructionBase: implement _with_new_tags

inducer · Jul 18, 2024 · da5c786 · da5c786
1 parent d50b720
commit da5c786
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/loopy/kernel/instruction.py b/loopy/kernel/instruction.py
@@ -463,6 +463,9 @@ def __setstate__(self, val):
         self.within_inames = (
                 intern_frozenset_of_ids(self.within_inames))
 
+    def _with_new_tags(self, tags: FrozenSet[Tag]):
+        return self.copy(tags=tags)
+
 # }}}