Function: use_stream_res_in_current_thread

Applicable Products

Product

Supported (√/x)

Ascend 950PR/Ascend 950DT

√

Atlas A3 training products/Atlas A3 inference products

√

Atlas A2 training products/Atlas A2 inference products

√

Atlas training products

√

Atlas inference products

√

Atlas 200I/500 A2 inference products

√

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 Values

Return Value

Description

ret

Int, 0 on success; else, failure.