Overview
ExeResGenerationContext is inherited from ExtendedKernelContext and is used to generate the context class of execution resources. It is mainly used to generate or configure various resources required for operator execution, including the memory, asynchronous execution flow, synchronization resources, operator attributes, and shape information of input and output tensors.
The following figure shows the inheritance relationship of ExeResGenerationContext.
Header Files to Be Included
1 | #include "exe_graph/runtime/exe_res_generation_context.h" |
Public Member Functions
ExecuteMode GetExecuteMode() const bool IsConstInput(const ge::AscendString &name) const const gert::StorageShape* GetInputShape(int64_t index) const const gert::StorageShape* GetOutputShape(int64_t index) const ge::graphStatus SetAttachedStreamInfos(std::vector<StreamInfo> &stream_info_vec) const std::vector<StreamInfo> GetAttachedStreamInfos() const int64_t GetStreamId() const ge::graphStatus SetSyncResInfos(std::vector<SyncResInfo> &sync_info_vec) const std::vector<SyncResInfo> GetSyncResInfos() const std::vector<int64_t> GetWorkspaceBytes() const void SetWorkspaceBytes(const std::vector<int64_t> &workspace_bytes) const int64_t GetOpId() const ge::graphStatus SetListStr(const std::string &attr_name, const std::vector<std::string> &list) const bool GetStrAttrVal(const char *attr_name, ge::AscendString &val) const bool SetStrAttrVal(const char *attr_name, const char *val) const bool GetIntAttrVal(const char *attr_name, int64_t &val) const bool SetIntAttrVal(const char *attr_name, const int64_t val) const
Parent topic: ExeResGenerationContext