GetStreamId

Applicability

Product

Supported

Atlas 350 Accelerator Card

Atlas A3 training product/Atlas A3 inference product

Atlas A2 training product/Atlas A2 inference product

Atlas 200I/500 A2 inference product

Atlas inference product

Atlas training product

Header File/Library File

  • Header file: #include <register/register_custom_pass.h>
  • Library file: libregister.so

Function Usage

Obtains the current stream ID on a node. The result indicates the stream ID allocated to the node after the built-in stream assignment algorithm is used.

Prototype

1
int64_t GetStreamId(const GNode &node) const

Parameters

Parameter

Input/Output

Description

node

Input

Node in the graph.

Returns

ID of the stream to which the node belongs.

Restrictions

None