Halide  12.0.1
Halide compiler and libraries
VectorizeLoops.h
Go to the documentation of this file.
1 #ifndef HALIDE_VECTORIZE_LOOPS_H
2 #define HALIDE_VECTORIZE_LOOPS_H
3 
4 /** \file
5  * Defines the lowering pass that vectorizes loops marked as such
6  */
7 
8 #include "Expr.h"
9 #include "Function.h"
10 
11 #include <map>
12 
13 namespace Halide {
14 
15 struct Target;
16 
17 namespace Internal {
18 
19 /** Take a statement with for loops marked for vectorization, and turn
20  * them into single statements that operate on vectors. The loops in
21  * question must have constant extent.
22  */
23 Stmt vectorize_loops(const Stmt &s, const std::map<std::string, Function> &env, const Target &t);
24 
25 } // namespace Internal
26 } // namespace Halide
27 
28 #endif
Base classes for Halide expressions (Halide::Expr) and statements (Halide::Internal::Stmt)
Defines the internal representation of a halide function and related classes.
Stmt vectorize_loops(const Stmt &s, const std::map< std::string, Function > &env, const Target &t)
Take a statement with for loops marked for vectorization, and turn them into single statements that o...
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