[FRONTEND] Remade exception picklable

2021-12-21 22:14:06 -08:00
parent a425f24d54
commit d8fce83e7a
1 changed files with 2 additions and 0 deletions
--- a/python/triton/code_gen.py
+++ b/python/triton/code_gen.py
@@ -544,6 +544,8 @@ class CompilationError(Exception):
        self.message += '\n'.join(src.split('\n')[:node.lineno])
        self.message += '\n' + ' ' * node.col_offset + '^'
        super().__init__(self.message)
+        # this is necessary to make CompilationError picklable
+        self.args = (src, node)


 class OutOfResources(Exception):