From 26cc77aa4ce9cb768f87681fede1ccc567239c9b Mon Sep 17 00:00:00 2001 From: Keren Zhou Date: Wed, 10 May 2023 13:22:17 -0400 Subject: [PATCH] [FRONTEND] Hotfix for contains_return_op `noinline` can be None, False, or True, so we have to check the callee in the first two cases --- python/triton/compiler/code_generator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/triton/compiler/code_generator.py b/python/triton/compiler/code_generator.py index 7d83de796033..17af1d8cc3c8 100644 --- a/python/triton/compiler/code_generator.py +++ b/python/triton/compiler/code_generator.py @@ -184,7 +184,7 @@ def check_undefined_name(cur_node): if check_undefined_name(node): return False fn = self.visit(node.func) - if isinstance(fn, JITFunction) and fn.noinline is False: + if isinstance(fn, JITFunction) and fn.noinline is not True: old_gscope = self.gscope self.gscope = sys.modules[fn.fn.__module__].__dict__ ret = self.contains_return_op(fn.parse())