昇腾社区首页
中文
注册
开发者
下载

简介

KernelLaunchInfo类为开发者提供创建和管理下发给Device的任务接口,允许开发者设置或者获取当前待下发Device任务的相关Launch信息。

本类可以支持用户创建新的Launch任务或者修改框架已经生成好的Launch任务。

需要包含的头文件

1
#include "graph/kernel_launch_info.h"

Public成员函数

static KernelLaunchInfo LoadFromData(const gert::ExeResGenerationContext *context, const std::vector<uint8_t> &data)
static KernelLaunchInfo CreateAicpuKfcTask(const gert::ExeResGenerationContext *context, const char *so_name, const char *kernel_name)
static KernelLaunchInfo CreateHcomRecordTask(const gert::ExeResGenerationContext *context, const char *group_name = "group")
static KernelLaunchInfo CreateHcomWaitTask(const gert::ExeResGenerationContext *context, const char *group_name = "group")
std::vector<uint8_t> Serialize()
uint32_t GetStreamId() const
void SetStreamId(uint32_t stream_id)
uint32_t GetBlockDim() const
graphStatus SetBlockDim(uint32_t block_dim)
const char *GetArgsFormat() const
graphStatus SetArgsFormat(const char *args_format)
const char *GetSoName() const
const char *GetKernelName() const