From f7ab96cfd754acd304bf83953ea33da3086e0b85 Mon Sep 17 00:00:00 2001
From: Philippe Tillet <phil@openai.com>
Date: Fri, 5 Nov 2021 09:26:33 -0700
Subject: [PATCH] [FRONTEND] Fixed some issues with `constexpr`

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

diff --git a/python/triton/code_gen.py b/python/triton/code_gen.py
index b8b9f8129..22b910f5a 100644
--- a/python/triton/code_gen.py
+++ b/python/triton/code_gen.py
@@ -152,7 +152,9 @@ class CodeGenerator(ast.NodeVisitor):
             if target in self.lscope:
                 raise ValueError(f'{target} is already defined.'
                                  f' constexpr cannot be reassigned.')
-            self.lscope[target] = triton.language.constexpr(value)
+            if not isinstance(value, triton.language.constexpr):
+                value = triton.language.constexpr(value)
+            self.lscope[target] = value
             return self.lscope[target]
         # default: call visit_Assign
         return self.visit_Assign(node)