tilus.Script.store_shared

tilus.Script.store_shared

Script.store_shared(dst, src, *, offsets=None, dims=None)[source]

Store a register tensor into a shared tensor.

This instruction stores the contents of the register tensor src into a slice of the shared tensor dst.

Parameters:

  • dst (SharedTensor) – The shared tensor to store into.

  • src (RegisterTensor) – The register tensor to store into the shared tensor.

  • offsets (Sequence[int], optional) – The offsets for each dimension of the shared tensor where the register tensor will be stored.

  • dims (Sequence[int], optional) – The dimensions of the shared tensor that are being sliced. If not provided, it is assumed that all dimensions are being sliced in the same order as the register tensor. The length of this sequence must match the number of dimensions of the register tensor being stored.

Return type:

None

Contents