[CODEGEN] Fixed issue with jit function passed as constexpr

2021-11-16 09:53:34 -08:00
parent 5b7ba3eb96
commit edd4b0c8b7
1 changed files with 2 additions and 0 deletions
--- a/python/triton/code_gen.py
+++ b/python/triton/code_gen.py
@@ -700,6 +700,8 @@ class Kernel:
            return 'B'
        if isinstance(obj, str):
            return 'str'
+        if isinstance(obj, JITFunction):
+            return ''
        assert False