From d8fce83e7a7d54ba66b55415b828d40e4f322169 Mon Sep 17 00:00:00 2001
From: Philippe Tillet <phil@openai.com>
Date: Tue, 21 Dec 2021 22:14:06 -0800
Subject: [PATCH] [FRONTEND] Remade exception picklable

---
 python/triton/code_gen.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/triton/code_gen.py b/python/triton/code_gen.py
index cacc39675..8393f2b87 100644
--- 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):