Collection of Schedule pass functions. More...

#include <tvm/te/schedule.h>
#include <tvm/tir/function.h>

Include dependency graph for schedule_pass.h:

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

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.

	tvm::te
	Tensor expression language DSL.

Functions
void	tvm::te::AutoInlineElemWise (Schedule sch)
	To automatically inline the element-wise operations. More...

void	tvm::te::AutoInlineBroarcast (Schedule sch)
	To automatically inline the broadcast operations. More...

void	tvm::te::AutoInlineInjective (Schedule sch)
	To automatically inline operations with injective writes (i.e. writes without reduction or sequential loops). Note that in this case, guarantees about contiguity, transpose, stride, alignemnt and memory footprint in general do not hold. More...

Map< IterVar, Range >	tvm::te::InferBound (const Schedule &sch)
	Infer the bound of all iteration variables relates to the schedule. More...

bool	tvm::te::VerifyCompactBuffer (const Stmt &stmt)
	Verify if there is any argument bound to compact buffer. More...

Stmt	tvm::te::ScheduleOps (Schedule s, Map< IterVar, Range > dom_map, bool debug_keep_trivial_loop)
	Schedule s' dependent operations. More...

PrimFunc	tvm::te::SchedulePostProcToPrimFunc (Array< ObjectRef > arg_list, Stmt body, Optional< Map< Tensor, Buffer >> bindings)
	Postprocessing the Stmt generated by ScheduleOps to create a PrimFunc that can then be used for further TIR optimizations. More...

Detailed Description

Collection of Schedule pass functions.

These passes works on the schedule hyper-graph and infers information such as bounds, check conditions read/write dependencies between the IterVar

Namespaces