Loading...
Searching...
No Matches
PartitionLoops.h
Go to the documentation of this file.
#ifndef PARTITION_LOOPS_H
#define PARTITION_LOOPS_H
/** \file
* Defines a lowering pass that partitions loop bodies into three
* to handle boundary conditions: A prologue, a simplified
* steady-stage, and an epilogue.
*/
#include "Expr.h"
#include "Scope.h"
namespace Halide {
namespace Internal {
/** Return true if an expression uses a likely tag that isn't captured by an
* enclosing Select, Min, or Max. The scope contains all vars that should be
* considered to have uncaptured likelies. */
bool has_uncaptured_likely_tag(const Expr &e, const Scope<> &scope);
/** Return true if an expression uses a likely tag. The scope contains all vars
* in scope that should be considered to have likely tags. */
bool has_likely_tag(const Expr &e, const Scope<> &scope);
/** Partitions loop bodies into a prologue, a steady state, and an
* epilogue. Finds the steady state by hunting for use of clamped
* ramps, or the 'likely' intrinsic. */
Stmt partition_loops(Stmt s);
} // namespace Internal
} // namespace Halide
#endif
Expr.h
Base classes for Halide expressions (Halide::Expr) and statements (Halide::Internal::Stmt).
Scope.h
Defines the Scope class, which is used for keeping track of names in a scope while traversing IR.
Halide::Internal::Scope
A common pattern when traversing Halide IR is that you need to keep track of stuff when you find a Le...
Definition Scope.h:94
Halide::Internal
Definition AbstractGenerator.h:24
Halide::Internal::has_likely_tag
bool has_likely_tag(const Expr &e, const Scope<> &scope)
Return true if an expression uses a likely tag.
Halide::Internal::partition_loops
Stmt partition_loops(Stmt s)
Partitions loop bodies into a prologue, a steady state, and an epilogue.
Halide::Internal::has_uncaptured_likely_tag
bool has_uncaptured_likely_tag(const Expr &e, const Scope<> &scope)
Return true if an expression uses a likely tag that isn't captured by an enclosing Select,...
Halide
This file defines the class FunctionDAG, which is our representation of a Halide pipeline,...
Definition AbstractGenerator.h:19
Halide::Expr
A fragment of Halide syntax.
Definition Expr.h:258
Halide::Internal::Stmt
A reference-counted handle to a statement node.
Definition Expr.h:427