Hi~ I think this is not the issue of tvmscript. For example, though List[Integer]
is supported by script, it would fail in lowering with Illegal attribute of key pragma_key, value type Array not supported
, since the annotation can not convert to an attr stmt.
import tvm
from tvm.script import tir as T
@T.prim_func
def fun(x: T.Buffer((16,), "int32")) -> None:
for k in T.serial(0, 16, annotations={"pragma_key": [1, 2, 3]}):
x[k] = 1
print(tvm.lower(fun).script())