GetHiddenInputSubType

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

Obtains the type of the hidden input address of ArgDescInfo. The correct value can be obtained only when type is kHiddenInput.

Prototype

HiddenInputSubType GetHiddenInputSubType() const

Parameters

None

Returns

Type of the hidden input address. The default value is kEnd.

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;
  aicpu_args_format.emplace_back(ArgDescInfo::CreateHiddenInput(HiddenInputSubType::kHcom));
  // kHcom is obtained.
  uint64_t hidden_type = aicpu_args_format.back().GetHiddenInputSubType();
...
}

Parent topic: ArgDescInfo