Files
apache--tvm/python
Eric Lunderberg d0b3ec93f9 [TVMScript] Allow T.Buffer[] arg annotation to use int as shape (#11454)
* [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
2022-05-30 16:13:50 +09:00
..