gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[FRONTEND] Fixed some issues with constexpr

Browse Source
This commit is contained in:
Philippe Tillet
2021-11-05 09:26:33 -07:00
parent 9a02dddf29
commit f7ab96cfd7
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/triton/code_gen.py
View File

@@ -152,7 +152,9 @@ class CodeGenerator(ast.NodeVisitor):
if target in self.lscope: if target in self.lscope:
raise ValueError(f'{target} is already defined.' raise ValueError(f'{target} is already defined.'
f' constexpr cannot be reassigned.') 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] return self.lscope[target]
# default: call visit_Assign # default: call visit_Assign
return self.visit_Assign(node) return self.visit_Assign(node)