Function: use_stream_res_in_current_thread

Applicability

Product

Supported

Atlas 350 Accelerator Card

Atlas A3 training product/Atlas A3 inference product

Atlas A2 training product/Atlas A2 inference product

Atlas training product

Atlas inference product

Atlas 200I/500 A2 inference product

Description

Sets the device resource limits of a specified stream in the current thread. If this API is called multiple times to set device resource limits for the same stream multiple times, the last setting takes effect.

Call acl.rt.set_stream_res_limit to set the device resource limit on the stream in advance.

Prototype

  • C Prototype
    aclError aclrtUseStreamResInCurrentThread(aclrtStream stream)
  • Python Function
    1
    ret = acl.rt.use_stream_res_in_current_thread(stream)
    

Parameters

Parameter

Description

stream

Int, specified stream. If 0 is passed, it indicates the default stream.

Return Value

Return Value

Description

ret

Int, 0 on success; else, failure.