Halide
BoundsInference.h
Go to the documentation of this file.
1 #ifndef HALIDE_BOUNDS_INFERENCE_H
2 #define HALIDE_BOUNDS_INFERENCE_H
3 
4 /** \file
5  * Defines the bounds_inference lowering pass.
6  */
7 
8 #include <map>
9 #include <string>
10 #include <vector>
11 
12 #include "Expr.h"
13 #include "Interval.h"
14 #include "Target.h"
15 
16 namespace Halide {
17 namespace Internal {
18 
19 class Function;
20 
21 /** Take a partially lowered statement that includes symbolic
22  * representations of the bounds over which things should be realized,
23  * and inject expressions defining those bounds.
24  */
25 Stmt bounds_inference(Stmt,
26  const std::vector<Function> &outputs,
27  const std::vector<std::string> &realization_order,
28  const std::vector<std::vector<std::string>> &fused_groups,
29  const std::map<std::string, Function> &environment,
30  const std::map<std::pair<std::string, int>, Interval> &func_bounds,
31  const Target &target);
32 
33 } // namespace Internal
34 } // namespace Halide
35 
36 #endif
Target.h
Halide
This file defines the class FunctionDAG, which is our representation of a Halide pipeline,...
Definition: AddAtomicMutex.h:21
Halide::LinkageType::Internal
@ Internal
Not visible externally, similar to 'static' linkage in C.
Halide::Internal::IOKind::Function
@ Function
Expr.h
Halide::Internal::realization_order
std::pair< std::vector< std::string >, std::vector< std::vector< std::string > > > realization_order(const std::vector< Function > &outputs, std::map< std::string, Function > &env)
Given a bunch of functions that call each other, determine an order in which to do the scheduling.
Interval.h
Halide::Internal::bounds_inference
Stmt bounds_inference(Stmt, const std::vector< Function > &outputs, const std::vector< std::string > &realization_order, const std::vector< std::vector< std::string >> &fused_groups, const std::map< std::string, Function > &environment, const std::map< std::pair< std::string, int >, Interval > &func_bounds, const Target &target)
Take a partially lowered statement that includes symbolic representations of the bounds over which th...