CreateCcuTask

Function Usage

Create a CCU task.

Prototype

static KernelLaunchInfo CreateCcuTask(const gert::ExeResGenerationContext *context, const std::vector<std::string>& groups)

Parameters

Parameter	Input/Output	Description
context	Input	Input parameter of the GenerateTask function, which stores the basic information about the operator.
groups	Input	Group name of the CCU task, which is a set of communicator names set by the operator.

Returns

Information about the created CCU task.

Constraints

None

Examples

graphStatus Mc2GenTaskCallback(const gert::ExeResGenerationContext *context,
    std::vector<std::vector<uint8_t>> &tasks) {
  ...
  //Assemble ccuGroups.
  std::vector<std::string> ccuGroups = {"group_ep", "group"};
  ge::KernelLaunchInfo ccuTask = ge::KernelLaunchInfo::CreateCcuTask(context, ccuGroups);
  ...
}

Parent topic: KernelLaunchInfo