GetHiddenInputSubType

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/arg_desc_info.h>
Library file: libgraph.so

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.

Restrictions

None

Example

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));
  // Obtain kHcom.
  uint64_t hidden_type = aicpu_args_format.back().GetHiddenInputSubType();
...
}

Parent topic: ArgDescInfo