npu.set_npu_loop_size

Product	Supported
Atlas A3 training products/Atlas A3 inference products	√
Atlas A2 training products/Atlas A2 inference products	√
Atlas 200I/500 A2 inference products	×
Atlas inference products	×
Atlas training products	√

Sets the number of iterations (or steps) per loop offloaded to the NPU.

npu.set_npu_loop_size(loop_size)

Parameter	Input/Output	Description
loop_size	Input	Number of steps per loop. Must be a positive integer.

None

import npu_device as npu
npu.set_npu_loop_size(100) # Set 100 steps to offload.