20 #ifndef TVM_META_SCHEDULE_POSTPROC_H_
21 #define TVM_META_SCHEDULE_POSTPROC_H_
23 #include <tvm/ffi/function.h>
24 #include <tvm/ffi/reflection/registry.h>
29 namespace meta_schedule {
96 using FAsString = ffi::TypedFunction<ffi::String()>;
Rules to apply a postprocessor to a schedule.
Definition: postproc.h:37
virtual void InitializeWithTuneContext(const TuneContext &context)=0
Initialize the design space generator with tuning context.
static constexpr const bool _type_mutable
Definition: postproc.h:66
virtual Postproc Clone() const =0
Clone the postprocessor.
virtual ~PostprocNode()=default
Virtual destructor.
virtual bool Apply(const tir::Schedule &sch)=0
Apply a postprocessor to the given schedule.
static void RegisterReflection()
Definition: postproc.h:42
TVM_FFI_DECLARE_OBJECT_INFO("meta_schedule.Postproc", PostprocNode, Object)
Managed reference to PostprocNode.
Definition: postproc.h:74
static Postproc DisallowAsyncStridedMemCopy()
Create a postprocessor that checks if all async mem copies are not strided.
static Postproc RewriteReductionBlock()
Create a postprocessor that rewrites reduction block by moving the init block out.
static Postproc RewriteUnboundBlock(int max_threadblocks)
Create a postprocessor that adds thread binding to unbound blocks.
static ffi::Array< Postproc, void > DefaultHexagon()
Create default postprocessors for Hexagon.
ffi::TypedFunction< ffi::String()> FAsString
Get the postprocessor function as string with name.
Definition: postproc.h:96
static ffi::Array< Postproc, void > DefaultCUDATensorCore()
Create default postprocessors for CUDA with TensorCore.
ffi::TypedFunction< void(const TuneContext &)> FInitializeWithTuneContext
The function type of InitializeWithTuneContext method.
Definition: postproc.h:80
ffi::TypedFunction< bool(const tir::Schedule &)> FApply
Apply a postprocessor to the given schedule.
Definition: postproc.h:86
static Postproc VerifyGPUCode()
Creates a postprocessor that verifies if the GPU code is correct.
static ffi::Array< Postproc, void > DefaultRISCV()
Create default postprocessors for RISCV.
static ffi::Array< Postproc, void > DefaultLLVM()
Create default postprocessors for LLVM.
static Postproc RewriteLayout()
Creates a postprocessor that rewrites the layout of input tensor.
static Postproc RewriteCooperativeFetch()
Create a postprocessor that rewrites the cooperative fetch annotation to actual vectorized cooperativ...
static Postproc RewriteTensorize(bool vectorize_init_loop=false)
Create a postprocessor that applies tensorization to annotated blocks.
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(Postproc, ObjectRef, PostprocNode)
static ffi::Array< Postproc, void > DefaultCPUTensorization()
Create default postprocessors for x86 (AVX512 and VNNI)
static ffi::Array< Postproc, void > DefaultCUDA()
Create default postprocessors for CUDA.
static Postproc VerifyVTCMLimit()
Verifies that the VTCM usage of a given schedule is within the provided limit.
static Postproc RewriteParallelVectorizeUnroll()
Creates a postprocessor that applies parallelization, vectorization and auto unrolling according to t...
ffi::TypedFunction< Postproc()> FClone
Clone the postprocessor.
Definition: postproc.h:91
static Postproc PyPostproc(FInitializeWithTuneContext f_initialize_with_tune_context, FApply f_apply, FClone f_clone, FAsString f_as_string)
Create a postprocessor with customized methods on the python-side.
static Postproc DisallowDynamicLoop()
Create a postprocessor that checks if all loops are static.
The postprocessor with customized methods on the python-side.
Definition: postproc.h:182
Postproc Clone() const final
Clone the postprocessor.
FAsString f_as_string
The packed function to the AsString function.
Definition: postproc.h:195
void InitializeWithTuneContext(const TuneContext &context) final
Initialize the design space generator with tuning context.
FApply f_apply
The packed function to the Apply function.
Definition: postproc.h:191
bool Apply(const tir::Schedule &sch) final
Apply a postprocessor to the given schedule.
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("meta_schedule.PyPostproc", PyPostprocNode, PostprocNode)
static void RegisterReflection()
Definition: postproc.h:197
FInitializeWithTuneContext f_initialize_with_tune_context
The packed function to the InitializeWithTuneContext function.
Definition: postproc.h:189
Postproc::FInitializeWithTuneContext FInitializeWithTuneContext
Definition: postproc.h:184
FClone f_clone
The packed function to the Clone function.
Definition: postproc.h:193
Postproc::FClone FClone
Definition: postproc.h:186
Postproc::FAsString FAsString
Definition: postproc.h:187
Postproc::FApply FApply
Definition: postproc.h:185
Managed reference to TuneContextNode.
Definition: tune_context.h:98
Managed reference to ScheduleNode.
Definition: schedule.h:885
Performance counters for profiling via the PAPI library.
Definition: analyzer.h:37
A managed object in the TVM runtime.