thread_bind.h File Reference

#include <tvm/tir/schedule/schedule.h>
#include <algorithm>
#include <limits>
#include <utility>

Include dependency graph for thread_bind.h:

Go to the source code of this file.

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

Functions
std::function< tir::ExprRV(int64_t)>	tvm::meta_schedule::MakeFactorSampler (tir::Schedule sch, Array< Integer > thread_extents)
	Given candidates of thread_extents, make a sampler that use sch->SampleCategorical to return a random thread extent. More...
Array< tir::LoopRV >	tvm::meta_schedule::BindSpatialLoop (tir::Schedule sch, tir::LoopRV loop, int64_t max_threadblocks, int64_t max_threads_per_block, std::function< tir::ExprRV(int64_t)> get_factor=nullptr)
	Bind blockIdx.x and threadIdx.x to the given loop. More...
void	tvm::meta_schedule::BindBlockThreadIdx (tir::Schedule sch, tir::BlockRV block, int64_t max_threadblocks, int64_t max_threads_per_block, std::function< tir::ExprRV(int64_t max_extent)> get_factor=nullptr)
	Bind the given block if it is not bound to blockIdx or threadIdx. More...