This document is relevant for: Inf2
, Trn1
, Trn2
nki.compiler.force_auto_alloc#
- nki.compiler.force_auto_alloc(func=None)[source]#
Force automatic allocation to be turned on in the kernel.
This will ignore any direct allocation inside the kernel
This document is relevant for: Inf2
, Trn1
, Trn2