SetArgsFormat

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 ArgsFormat for an operator.

Prototype

      
           graphStatus SetArgsFormat(const char *args_format)

Parameters

Parameter	Input/Output	Description
args_format	Input	ArgsFormat of the operator.

Returns

Parameter	Type	Description
-	graphStatus	GRAPH_SUCCESS(0): success. Other values: failure. For details, see ge::graphStatus.

Parameter

Type

Description

graphStatus

GRAPH_SUCCESS(0): success.

Other values: failure. For details, see ge::graphStatus.

Restrictions

ArgsFormat is only required for AI CPU and AI Core tasks.

Examples

      
       
         
         
           graphStatus Mc2GenTaskCallback(const gert::ExeResGenerationContext *context,
    std::vector<std::vector<uint8_t>> &tasks) {
  ...
  auto aicore_task = KernelLaunchInfo::LoadFromData(context, tasks.back());
  auto aicore_args_format_str = aicore_task.GetArgsFormat();
  auto aicore_args_format = ArgsFormatSerializer::Deserialize(aicore_args_format_str);
  size_t i = 0UL;
  for (; i < aicore_args_format.size(); i++) {
    if (aicore_args_format[i].GetType() == ArgDescType::kIrInput ||
        aicore_args_format[i].GetType() == ArgDescType::kInputInstance) {
      break;
    }
  }
  // Insert HiddenInput.
  aicore_args_format.insert(aicore_args_format.begin() + i, ArgDescInfo::CreateHiddenInput(HiddenInputSubType::kHcom));
  aicore_task.SetArgsFormat(ArgsFormatSerializer::Serialize(aicore_args_format).GetString());
  ...
}

          

        

      
     

Parent topic: KernelLaunchInfo