Halide  12.0.1
Halide compiler and libraries
StorageFlattening.h
Go to the documentation of this file.
1 #ifndef HALIDE_STORAGE_FLATTENING_H
2 #define HALIDE_STORAGE_FLATTENING_H
3 
4 /** \file
5  * Defines the lowering pass that flattens multi-dimensional storage
6  * into single-dimensional array access
7  */
8 
9 #include <map>
10 #include <string>
11 #include <vector>
12 
13 #include "Expr.h"
14 
15 namespace Halide {
16 
17 struct Target;
18 
19 namespace Internal {
20 
21 class Function;
22 
23 /** Take a statement with multi-dimensional Realize, Provide, and Call
24  * nodes, and turn it into a statement with single-dimensional
25  * Allocate, Store, and Load nodes respectively. */
27  const std::vector<Function> &outputs,
28  const std::map<std::string, Function> &env,
29  const Target &target);
30 
31 } // namespace Internal
32 } // namespace Halide
33 
34 #endif
Base classes for Halide expressions (Halide::Expr) and statements (Halide::Internal::Stmt)
Stmt storage_flattening(Stmt s, const std::vector< Function > &outputs, const std::map< std::string, Function > &env, const Target &target)
Take a statement with multi-dimensional Realize, Provide, and Call nodes, and turn it into a statemen...
This file defines the class FunctionDAG, which is our representation of a Halide pipeline,...
@ Internal
Not visible externally, similar to 'static' linkage in C.
A reference-counted handle to a statement node.
Definition: Expr.h:413
A struct representing a target machine and os to generate code for.
Definition: Target.h:19