diff --git a/python/triton/code_gen.py b/python/triton/code_gen.py
index 236723057..1c28d0b4e 100644
--- a/python/triton/code_gen.py
+++ b/python/triton/code_gen.py
@@ -67,6 +67,7 @@ class CodeGenerator(ast.NodeVisitor):
             'float': float,
             'int': int,
             'print': print,
+            'isinstance': isinstance,
             'getattr': getattr,
         }