SetStreamId

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 <graph/kernel_launch_info.h>
Library file: libregister.so

Function Usage

Sets the stream ID for a task.

Prototype

void SetStreamId(uint32_t stream_id)

Parameters

Parameter	Input/Output	Description
stream_id	Input	Stream ID.

Returns

None

Restrictions

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