reduction.h File Reference

CUDA schedule for reduction operations. More...

#include <tvm/target/generic_func.h>
#include <tvm/te/operation.h>
#include <tvm/te/schedule_pass.h>
#include <tvm/topi/detail/fuse.h>
#include <tvm/topi/tags.h>

Include dependency graph for reduction.h:

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

Go to the source code of this file.

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::topi
	tvm::topi::cuda

Functions
Schedule	tvm::topi::cuda::ScheduleReduce (const Target &target, Operation op, Schedule sch, bool is_idx_reduce=false)
	Schedule a given reduce operation. More...
void	tvm::topi::cuda::TraverseBeforeReduce (Schedule s, Operation op)
	Recursively traverse operator inputs, setting injective inputs to be computed inline. More...
void	tvm::topi::cuda::TraverseAfterReduce (const Target &target, Schedule s, Operation op)
	Schedule a reduce op, then invoke TraverseBeforeReduce on each of the op's inputs. More...
Schedule	tvm::topi::cuda::schedule_reduce (const Target &target, Array< Tensor > outs)
	Create a CUDA schedule for a reduce operation. More...

Detailed Description

CUDA schedule for reduction operations.