nki.language.psum — AWS Neuron Documentation (original) (raw)

This document is relevant for: Inf2, Trn1, Trn2

nki.language.psum#

nki.language.psum = Ellipsis#

PSUM - Only visible to each individual kernel instance in the SPMD grid, alias of nki.compiler.psum.auto_alloc()

This document is relevant for: Inf2, Trn1, Trn2