SetStreamId

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	√
Atlas inference products	√
Atlas training products	√

Header File

#include <graph/kernel_launch_info.h>

Function Usage

Sets the stream ID of a task.

Prototype

void SetStreamId(uint32_t stream_id)

Parameters

Parameter	Input/Output	Description
stream_id	Input	Stream ID.

Returns

None

Constraints

None

Examples

graphStatus Mc2GenTaskCallback(const gert::ExeResGenerationContext *context,
    std::vector<std::vector<uint8_t>> &tasks) {
  ...
  auto wait_task = KernelLaunchInfo::CreateHcomWaitTask(context);
  wait_task.SetStreamId(attach_stream_id);
  ...
}

Parent topic: KernelLaunchInfo