From 22105bc33b14ce4f9af669a50b6031bdb6ebb8c4 Mon Sep 17 00:00:00 2001 From: vesuppi Date: Sun, 3 Jul 2022 15:25:37 -0700 Subject: [PATCH] [FRONTEND] Added type check in semantic arange (#572) --- python/triton/language/semantic.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/python/triton/language/semantic.py b/python/triton/language/semantic.py index 9025319d6..a31fec384 100644 --- a/python/triton/language/semantic.py +++ b/python/triton/language/semantic.py @@ -436,6 +436,9 @@ def not_equal(input: tl.tensor, def arange(start: int, end: int, builder: ir.builder) -> tl.tensor: + if not isinstance(start, int) or not isinstance(end, int): + raise ValueError("arange's arguments must be of type tl.constexpr") + shape = [end - start] ret_ty = tl.block_type(tl.int32, shape) return tl.tensor(builder.get_range(start, end), ret_ty)