Halide
AllocationBoundsInference.h
Go to the documentation of this file.
1 #ifndef HALIDE_ALLOCATION_BOUNDS_INFERENCE_H
2 #define HALIDE_ALLOCATION_BOUNDS_INFERENCE_H
3 
4 /** \file
5  * Defines the lowering pass that determines how large internal allocations should be.
6  */
7 #include <map>
8 #include <string>
9 #include <utility>
10 
11 #include "Expr.h"
12 #include "Interval.h"
13 
14 namespace Halide {
15 namespace Internal {
16 
17 class Function;
18 
19 /** Take a partially statement with Realize nodes in terms of
20  * variables, and define values for those variables. */
21 Stmt allocation_bounds_inference(Stmt s,
22  const std::map<std::string, Function> &env,
23  const std::map<std::pair<std::string, int>, Interval> &func_bounds);
24 } // namespace Internal
25 } // namespace Halide
26 
27 #endif
Halide::Internal::allocation_bounds_inference
Stmt allocation_bounds_inference(Stmt s, const std::map< std::string, Function > &env, const std::map< std::pair< std::string, int >, Interval > &func_bounds)
Take a partially statement with Realize nodes in terms of variables, and define values for those vari...
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
Interval.h