Halide 22.0.0 Halide compiler and libraries

ApplySplit.h
Go to the documentation of this file.
1#ifndef APPLY_SPLIT_H
2#define APPLY_SPLIT_H
3
4/** \file
5 *
6 * Defines method that returns a list of let stmts, substitutions, and
7 * predicates to be added given a split schedule.
8 */
9
10#include <map>
11#include <string>
12#include <utility>
13#include <vector>
14
15#include "Expr.h"
16#include "Schedule.h"
17
18namespace Halide {
19namespace Internal {
20
21 struct ApplySplitResult {
22 // If type is "Substitution", then this represents a substitution of
23 // variable "name" to value. ProvideValueSubstitution and
24 // ProvideArgSubstitution are similar, but only apply to instances
25 // found on the LHS or RHS of a call or provide. respectively. If
26 // type is "LetStmt", we should insert a new let stmt defining
27 // "name" with value "value". If type is "Predicate", we should
28 // ignore "name" and the predicate is "value".
29
30 std::string name;
31 Expr value;
32
33 enum Type { Substitution = 0,
34 SubstitutionInCalls,
35 SubstitutionInProvides,
36 LetStmt,
37 PredicateCalls,
38 PredicateProvides,
39 Predicate,
40 BlendProvides };
41 Type type;
42
43 ApplySplitResult(const std::string &n, Expr val, Type t)
44 : name(n), value(std::move(val)), type(t) {
45 }
46 ApplySplitResult(Expr val, Type t = Predicate)
47 : name(""), value(std::move(val)), type(t) {
48 }
49};
50
51/** Given a Split schedule on a definition (init or update), return a list of
52 * of predicates on the definition, substitutions that needs to be applied to
53 * the definition (in ascending order of application), and let stmts which
54 * defined the values of variables referred by the predicates and substitutions
55 * (ordered from innermost to outermost let). */
56std::vector<ApplySplitResult> apply_split(
57 const Split &split, const std::string &prefix,
58 std::map<std::string, Expr> &dim_extent_alignment);
59
60/** Compute the loop bounds of the new dimensions resulting from applying the
61 * split schedules using the loop bounds of the old dimensions. */
62std::vector<std::pair<std::string, Expr>> compute_loop_bounds_after_split(
63 const Split &split, const std::string &prefix);
64
65} // namespace Internal
66} // namespace Halide
67
68#endif
Expr.h
Base classes for Halide expressions (Halide::Expr) and statements (Halide::Internal::Stmt)
Schedule.h
Defines the internal representation of the schedule for a function.
Halide::Internal::apply_split
std::vector< ApplySplitResult > apply_split(const Split &split, const std::string &prefix, std::map< std::string, Expr > &dim_extent_alignment)
Given a Split schedule on a definition (init or update), return a list of of predicates on the defini...
Halide::Internal::compute_loop_bounds_after_split
std::vector< std::pair< std::string, Expr > > compute_loop_bounds_after_split(const Split &split, const std::string &prefix)
Compute the loop bounds of the new dimensions resulting from applying the split schedules using the l...
Halide
This file defines the class FunctionDAG, which is our representation of a Halide pipeline,...
Definition: AbstractGenerator.h:19
Halide::LinkageType::Internal
@ Internal
Not visible externally, similar to 'static' linkage in C.
Halide::Expr
A fragment of Halide syntax.
Definition: Expr.h:258
Halide::Internal::ApplySplitResult
Definition: ApplySplit.h:21
Halide::Internal::ApplySplitResult::ApplySplitResult
ApplySplitResult(Expr val, Type t=Predicate)
Definition: ApplySplit.h:46
Halide::Internal::ApplySplitResult::Type
Type
Definition: ApplySplit.h:33
Halide::Internal::ApplySplitResult::Substitution
@ Substitution
Definition: ApplySplit.h:33
Halide::Internal::ApplySplitResult::PredicateProvides
@ PredicateProvides
Definition: ApplySplit.h:38
Halide::Internal::ApplySplitResult::LetStmt
@ LetStmt
Definition: ApplySplit.h:36
Halide::Internal::ApplySplitResult::Predicate
@ Predicate
Definition: ApplySplit.h:39
Halide::Internal::ApplySplitResult::PredicateCalls
@ PredicateCalls
Definition: ApplySplit.h:37
Halide::Internal::ApplySplitResult::BlendProvides
@ BlendProvides
Definition: ApplySplit.h:40
Halide::Internal::ApplySplitResult::SubstitutionInCalls
@ SubstitutionInCalls
Definition: ApplySplit.h:34
Halide::Internal::ApplySplitResult::SubstitutionInProvides
@ SubstitutionInProvides
Definition: ApplySplit.h:35
Halide::Internal::ApplySplitResult::name
std::string name
Definition: ApplySplit.h:30
Halide::Internal::ApplySplitResult::type
Type type
Definition: ApplySplit.h:41
Halide::Internal::ApplySplitResult::ApplySplitResult
ApplySplitResult(const std::string &n, Expr val, Type t)
Definition: ApplySplit.h:43
Halide::Internal::ApplySplitResult::value
Expr value
Definition: ApplySplit.h:31
Halide::Internal::Split
Definition: Schedule.h:325