[FRONTEND] fixup (#505)

2022-04-30 14:25:06 -07:00
parent 11a908655d
commit 95feb10ec9
1 changed files with 2 additions and 0 deletions
--- 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