From edd4b0c8b7e99271023384e52c8cdbc913328235 Mon Sep 17 00:00:00 2001 From: Philippe Tillet Date: Tue, 16 Nov 2021 09:53:34 -0800 Subject: [PATCH] [CODEGEN] Fixed issue with jit function passed as constexpr --- 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 cc0a103c9..1e514d3e2 100644 --- 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