[HYRID][SYNTAX] tir.allocate

[tqchen]

Right now Allocate is printed as

def func():
    data = tir.var("handle")
    tir.allocate(data, "float32", [1024])

After apache/tvm#6317, we want to make sure that allocate actually uses var that has a pointer type annotation(non runtime type) to get richer information, it would be great to enhance hybrid parser to support this case.

A slight generalization might look like

def func():
    data = tir.var(ty.Ptr[ty.float32])
    tir.allocate(data, "float32", [1024])

However, it seems to be weird to separate data var declaration. In this case, we can simply do

def func():
    # data is a var with type annotation "Ptr[float32["
    data = tir.allocate("float32", [1024])  

and in the case of scoping

def func():
    with tir.allocate("float32", [1024]) as data:

[tqchen]

cc @spectrometerHBH

[spectrometerHBH]

I encounter a problem now

The format now is

packedB = tir.var("handle")
tir.attr(packedB, "storage_scope", "global")
tir.allocate(packedB, "float32x32", [32768])

After compressing tir.allocate

tir.attr(packedB, "storage_scope", "global")
packedB = tir.allocate("float32x32", [32768])

From python's point of view, packedB is used before its declaration.

cc @tqchen

[Hzfengsy]

Can we compress tir.allocate with attr together? As far as I know, every allocate will have an attr above it.

So, it could be

packedB = tir.allocate("float32x32", [32768], "global")

[tqchen]

I agree in this case we should try to fold the storage scope into the allocate

[tqchen]

@spectrometerHBH can you work a bit to resolve this issue? Thanks!