From 95feb10ec9538ab225c04b2b4b7b0f6b777a88cf Mon Sep 17 00:00:00 2001 From: Philippe Tillet Date: Sat, 30 Apr 2022 14:25:06 -0700 Subject: [PATCH] [FRONTEND] fixup (#505) --- python/triton/language/core.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/triton/language/core.py b/python/triton/language/core.py index 046e60e0e..9af259a14 100644 --- a/python/triton/language/core.py +++ b/python/triton/language/core.py @@ -27,6 +27,8 @@ def _to_tensor(x, builder): elif isinstance(x, float): return tensor(builder.get_float32(x), float32) elif isinstance(x, constexpr): + if x.value is None: + return None return _to_tensor(x.value, builder) elif isinstance(x, tensor): return x