GEStreamAllocationSummaryGetHcclFollowedStreamNums

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

x

Atlas inference product

Atlas training product

Header File/Library File

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

Function Usage

Obtains the actual number of physical streams of the communication secondary stream corresponding to the root graph and subgraphs.

Prototype

1
ge::Status GEStreamAllocationSummaryGetHcclFollowedStreamNums(const ge::CompiledGraphSummary &compiled_graph_summary, std::map<AscendString, std::vector<int64_t>> &graph_to_hccl_followed_stream_nums);

Parameters

Parameter

Input/Output

Description

compiled_graph_summary

Input

Brief information about the graph build result.

graph_to_hccl_followed_stream_nums

Output

A value in Map format. The key is the graph name. The value is the vector indicating the actual number of physical streams of the communication secondary stream. 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.

Restrictions

None