function.h File Reference

TIR Function. More...

#include <tvm/ffi/container/map.h>
#include <tvm/ffi/container/variant.h>
#include <tvm/ir/cow.h>
#include <tvm/ir/function.h>
#include <tvm/ir/prim/expr.h>
#include <tvm/runtime/tensor.h>
#include <tvm/tirx/buffer.h>
#include <tvm/tirx/stmt.h>
#include <string>

Include dependency graph for function.h:

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

Go to the source code of this file.

Classes
class	tvm::tirx::PrimFuncNode
	Primitive functions that contains TIR statements. More...
class	tvm::tirx::PrimFunc
	Managed reference to PrimFuncNode. More...
class	tvm::tirx::TensorIntrinNode
	Tensor intrinsics for tensorization. More...
class	tvm::tirx::TensorIntrin
	Managed reference to TensorIntrinNode. More...

Namespaces
namespace	tvm
	An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
namespace	tvm::tirx
namespace	tvm::tirx::attr
	PrimFunc specific attribute names.

Functions
PrimFunc	tvm::tirx::Specialize (PrimFunc func, const ffi::Map< Var, ffi::Variant< BufferVar, Expr > > &param_map)
	Specialize parameters of PrimFunc.

Variables
constexpr const char *	tvm::tirx::attr::kKernelLaunchParams = "tirx.kernel_launch_params"
	List of thread IterVar that a DeviceLaunch function corresponds to.
constexpr const char *	tvm::tirx::attr::kLaunchBoundsMinBlocksPerSM = "tirx.launch_bounds_min_blocks_per_sm"
	CUDA launch bound minimum CTAs per SM.
constexpr const char *	tvm::tirx::attr::kLaunchBoundsMaxBlocksPerCluster
	CUDA launch bound maximum CTAs per cluster.
constexpr const char *	tvm::tirx::attr::kMaxRegisters = "tirx.max_registers"
	CUDA maximum registers per thread.
constexpr const char *	tvm::tirx::attr::kRequiredBlockSize = "tirx.required_block_size"
	Require CUDA to use the statically-declared block and cluster dimensions.
constexpr const char *	tvm::tirx::attr::kNoAlias = "tirx.noalias"
	Whether to set noalias rule on the function arguments.
constexpr const char *	tvm::tirx::attr::kIsEntryFunc = "tirx.is_entry_func"
	Mark the function as the entry function of the final generated runtime module.
constexpr const char *	tvm::tirx::attr::kIsGlobalFunc = "tirx.is_global_func"
	Mark the function as the global function called from the host.
constexpr const char *	tvm::tirx::attr::kIsHostFunc = "tirx.is_host_func"
	Mark the function as run on the host, mutually exclusive with kTarget.
constexpr const char *	tvm::tirx::attr::kIsScheduled = "tirx.is_scheduled"
	Mark the function as scheduled, so the default schedule will pass will skip it.

Detailed Description

TIR Function.