GEStreamAllocationSummaryIsAssignedByStreamPass

Applicability

Product

Supported or Not

Atlas A3 training products/Atlas A3 inference products

Atlas A2 training products/Atlas A2 inference products

Atlas 200I/500 A2 inference products

x

Atlas inference products

Atlas training products

Header File/Library File

  • Header file: #include <ge/ge_graph_compile_summary.h>
  • Library file: libge_compiler.so

Function Usage

Checks whether the root graph and subgraph are allocated by the stream allocation pass registered by the user.

Prototype

1
ge::Status GEStreamAllocationSummaryIsAssignedByStreamPass(const ge::CompiledGraphSummary &compiled_graph_summary, std::map<AscendString, std::vector<bool>> &graph_to_is_assigned_by_stream_pass)

Parameters

Parameter

Input/Output

Description

compiled_graph_summary

Input

Brief information about the graph build result.

graph_to_is_assigned_by_stream_pass

Output

A value in Map format. The key is the graph name. The value is the vector allocated by stream allocation pass. The index indicates the logical stream ID.

Returns

Parameter

Type

Description

-

ge::Status

  • SUCCESS: The API is called successfully.
  • FAILED: The API fails to be called.

Constraints

None