state.h File Reference

This file defines ScheduleState, the core data structure of TensorIR scheduling. More...

#include <tvm/ir/module.h>
#include <tvm/tir/block_scope.h>
#include <tvm/tir/function.h>
#include <unordered_map>
#include <utility>

Include dependency graph for state.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	tvm::tir::BlockInfo
	The information about a TensorIR block, it contains two categories of information 1) Info on the block scope rooted at a specific block, including dependency tracking, flags indicating if the scope is a stage pipeline, etc. 2) Info on the block itself, including if the block has a quasi-affine binding, if the regions it reads are completely covered by their producers, etc. More...
class	tvm::tir::ScheduleStateNode
	The state of scheduling, which exposes a Replace method as the primary interface for all the scheduling primitives to manipulate the TensorIR. More...
class	tvm::tir::ScheduleState
	Managed reference to ScheduleStateNode. More...

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::tir

Enumerations
enum	tvm::tir::ScheduleDebugMask : uint32_t { tvm::tir::kVerifySRefTree = 1 , tvm::tir::kVerifyCachedFlags = 2 }
	The bitmask of the debug flag in the ScheduleStateNode. More...

Detailed Description

This file defines ScheduleState, the core data structure of TensorIR scheduling.