昇腾社区首页
中文
注册

函数:set_stream_attribute

产品支持情况

产品

是否支持

Atlas A3 训练系列产品/Atlas A3 推理系列产品

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

Atlas 训练系列产品

Atlas 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 200/300/500 推理产品

功能说明

设置Stream属性值。

函数原型

  • C函数原型
    aclError aclrtSetStreamAttribute(aclrtStream stream, aclrtStreamAttr stmAttrType, aclrtStreamAttrValue *value)
  • python函数
    1
    ret = acl.rt.set_stream_attribute(stream, stm_attr_type, value)
    

参数说明

参数名

说明

stream

int,指定Stream。若传入0,则操作默认Stream。

stm_attr_type

int,属性类型具体请参见新增数据结构aclrtStreamAttr

value

dict,属性值。具体请参见新增数据结构aclrtStreamAttrValue

返回值说明

返回值

说明

ret

int,返回0表示成功,返回其他值表示失败。

约束说明

  • 溢出检测属性:调用该接口打开或关闭溢出检测开关后,仅对后续新下的任务生效,已下发的任务仍维持原样。
  • Failure Mode:不支持对Context默认Stream设置Failure Mode。
  • stm_attr_type设置为ACL_STREAM_ATTR_FAILURE_MODE时,支持芯片为310P、310B、910、910B、910C
  • stm_attr_type设置为ACL_STREAM_ATTR_FLOAT_OVERFLOW_CHECK、ACL_STREAM_ATTR_USER_CUSTOM_TAG时,支持芯片为910B、910C