InputAddr

产品支持情况

全量芯片支持。

头文件/库文件

• 头文件：#include <exe_graph/runtime/annotated_args_context.h>
• 库文件：liblowering.so

功能说明

gert::InputAddr是逻辑输入地址描述符，作为AnnotatedKernelArgs::AppendArg的参数，用于把kernel args中的一个参数标记为算子输入地址。

函数原型

namespace gert {
struct InputAddr {
  uint32_t index;    // 算子实例中的输入index
  const void *addr;  // 输入逻辑地址
};
}  // namespace gert

参数说明

参数名	类型	说明
index	uint32_t	算子实例中的输入index。值必须不大于INT32_MAX。
addr	const void *	追加到args参数的输入逻辑地址。

返回值说明

无

约束说明

• addr应来自当前编译期回调可用的输入Tensor，例如AnnotatedArgsContext::GetInputTensor返回值的地址。

调用示例

const auto *input = ctx.GetInputTensor(0U);
if (input == nullptr) {
  return ge::GRAPH_FAILED;
}
gert::AnnotatedKernelArgs args(
    gert::InputAddr{0U, input->GetAddr()});
return ctx.AddLaunch(launch_info, std::move(args));