SetHiddenInputSubType

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/arg_desc_info.h>

Function Usage

Sets the type of the hidden input address of ArgDescInfo. The setting is successful only when type is kHiddenInput.

Prototype

      
           graphStatus SetHiddenInputSubType(HiddenInputSubType hidden_type)

Parameters

Parameter	Input/Output	Description
hidden_type	Input	Hidden input type.

Returns

ge::GRAPH_SUCCESS on success.

For details about the definition of graphStatus, see ge::graphStatus.

Constraints

None

Examples

      
           graphStatus Mc2GenTaskCallback(const gert::ExeResGenerationContext *context,
    std::vector<std::vector<uint8_t>> &tasks) {
...
  // Set an AI CPU task.
  auto aicpu_task = KernelLaunchInfo::CreateAicpuKfcTask(context,
      "libccl_kernel.so", "RunAicpuKfcSrvLaunch");
  std::vector<ArgDescInfo> aicpu_args_format;
  ArgDescInfo args_info(ArgDescType::kHiddenInput);
  args_info.SetHiddenInputSubType(HiddenInputSubType::kHcom);
  aicpu_args_format.emplace_back(args_info);
...
}

Parent topic: ArgDescInfo