nki.compiler.force_auto_alloc — AWS Neuron Documentation (original) (raw)

Contents

• force_auto_alloc()

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