昇腾社区首页
中文
注册

npu.set_npu_loop_size

产品支持情况

产品

是否支持

Atlas A2 训练系列产品

Atlas 800I A2 推理产品

x

Atlas 200I/500 A2 推理产品

x

Atlas 推理系列产品

x

Atlas 训练系列产品

Atlas 200/300/500 推理产品

x

功能说明

用于设置NPU循环下沉执行时的循环次数。

函数原型

1
npu.set_npu_loop_size(loop_size)

参数说明

参数名

输入/输出

描述

loop_size

输入

NPU循环下沉执行时的循环次数,必须为大于0的整数。

返回值

无。

调用示例

1
2
import npu_device as npu
npu.set_npu_loop_size(100) # 设置循环下沉次数为100