Halide 22.0.0 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
13namespace Halide {
14
15struct Target;
16
17namespace 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
Expr.h
Base classes for Halide expressions (Halide::Expr) and statements (Halide::Internal::Stmt)
Function.h
Defines the internal representation of a halide function and related classes.
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.
Halide::Internal::Stmt
A reference-counted handle to a statement node.
Definition: Expr.h:427