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