aclrtSetDeviceResLimit

产品	是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件	√
Atlas 200I/500 A2 推理产品	√
Atlas 推理系列产品	√
Atlas 训练系列产品	√
Atlas 200/300/500 推理产品	√

设置当前进程的Device资源限制。

本接口应在调用aclrtSetDevice接口之后且在执行算子之前使用。如果对同一Device进行多次设置，将以最后一次设置为准。

aclError aclrtSetDeviceResLimit(int32_t deviceId, aclrtDevResLimitType type, uint32_t value)

参数名	输入/输出	说明
deviceId	输入	Device ID。用户调用aclrtGetDeviceCount接口获取可用的Device数量后，这个Device ID的取值范围：[0, (可用的Device数量-1)]
type	输入	资源类型，当前支持Cube（矩阵）计算单元、Vector（向量）计算单元。
value	输入	资源限制的大小。

参数名

输入/输出

说明

deviceId

输入

Device ID。

用户调用aclrtGetDeviceCount接口获取可用的Device数量后，这个Device ID的取值范围：[0, (可用的Device数量-1)]

type

输入

资源类型，当前支持Cube（矩阵）计算单元、Vector（向量）计算单元。

value

输入

资源限制的大小。

返回0表示成功，返回其他值表示失败，请参见aclError。

本接口的设置仅对后续下发的任务有效。例如在调用aclmdlRICaptureBegin、aclmdlRICaptureEnd等接口捕获Stream任务到模型中、再执行模型推理的场景下，则需要在捕获之前调用本接口设置Device资源。

父主题： Device管理