This document is relevant for: Trn2, Trn3

nki.language.store#

nki.language.store(dst, value)[source]#

Store into a tensor on device memory (HBM) from on-chip memory (SBUF).

Warning

This API is experimental and may change in future releases.

Parameters:

  • dst – HBM tensor to store the data into.

  • value – an SBUF tile that contains the values to store.

This document is relevant for: Trn2, Trn3