nki.compiler.sbuf.auto_alloc

nki.compiler.sbuf.auto_alloc()

Returns a maker to indicate the tensor should be automatically allocated by compiler. All SBUF tensors in a kernel must either all be marked as auto_alloc(), or all be allocated with alloc or mod_alloc.

Initialize a tensor with buffer=nl.sbuf is equivalent to buffer=ncc.sbuf.auto_alloc().