d0b3ec93f9
* [TVMScript] Allow T.Buffer[] arg annotation to use int as shape Both the function `tvm.tir.decl_buffer` and the TVMScript `T.match_buffer` expression allow a `PrimExpr` to be passed as the buffer shape, which is interpreted as a 1-d buffer of that size. This allows the same behavior to be used in the `T.Buffer` syntactic sugar. (e.g. `A: T.Buffer[16, "float32"]` instead of `A: T.Buffer[(16,), "float32"`) * Fixed round-trip when buffer size contains an expression