Halide
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);
24
25
}
// namespace Internal
26
}
// namespace Halide
27
28
#endif
Halide::Internal::vectorize_loops
Stmt vectorize_loops(const Stmt &s, const std::map< std::string, Function > &env)
Take a statement with for loops marked for vectorization, and turn them into single statements that o...
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.
Expr.h
Function.h
src
VectorizeLoops.h
Generated by
1.8.17