Overview
TilingContext is inherited from ExtendedKernelContext. It is a context class used for operator tiling computation and provides APIs for obtaining operator compilation information (CompileInfo), input and output shapes, input tensor values, and APIs for accessing operator tiling parameters. In operator tiling implementation, you can use this class to obtain the parameters on which tiling computation depends, and save the tiling result (for example, TilingKey and TilingData) to the context for subsequent operator execution.
The following figure shows the inheritance relationship of TilingContext.
Header Files to Be Included
1 | #include <tiling_context.h> |
Public Member Functions
const StorageShape *GetInputShape(const size_t index) const const Tensor *GetInputTensor(const size_t index) const const Tensor *GetOptionalInputTensor(const size_t ir_index) const const Tensor *GetDynamicInputTensor(const size_t ir_index, const size_t relative_index) const const Tensor *GetRequiredInputTensor(const size_t ir_index) const const StorageShape *GetOptionalInputShape(const size_t ir_index) const const StorageShape *GetDynamicInputShape(const size_t ir_index, const size_t relative_index) const const StorageShape *GetRequiredInputShape(const size_t ir_index) const const StorageShape *GetOutputShape(size_t index) const ge::graphStatus SetTilingKey(const uint64_t tiling_key) uint64_t GetTilingKey() const ge::graphStatus SetBlockDim(const uint32_t block_dim) uint32_t GetBlockDim() const ge::graphStatus SetTilingCond(int32_t tiling_cond) int32_t GetTilingCond() const ge::graphStatus SetNeedAtomic(const bool atomic) bool NeedAtomic() const template<typename T> auto GetTilingData() -> T* TilingData *GetRawTilingData() size_t *GetWorkspaceSizes(const size_t workspace_count) size_t GetWorkspaceNum() const fe::PlatFormInfos *GetPlatformInfo() const template<typename T> const T *GetCompileInfo() const const void *GetCompileInfo() const ge::graphStatus SetScheduleMode(const uint32_t schedule_mode) uint32_t GetScheduleMode() const ge::graphStatus SetLocalMemorySize(const uint32_t local_memory_size) uint32_t GetLocalMemorySize() int32_t GetDeterministic() const
Parent topic: TilingContext