Revision 3fbb12aa928a87a4c4e69a95eaf6f5ee02958600 authored by John Lawson on 09 February 2021, 18:00:02 UTC, committed by GitHub on 09 February 2021, 18:00:02 UTC
* Add support for AVX512 BF16 dot product * Match on f32*f32 * Remove f32 check
1 parent 3e034d6
VectorizeLoops.h
#ifndef HALIDE_VECTORIZE_LOOPS_H
#define HALIDE_VECTORIZE_LOOPS_H
/** \file
* Defines the lowering pass that vectorizes loops marked as such
*/
#include "Expr.h"
#include "Function.h"
#include <map>
namespace Halide {
struct Target;
namespace Internal {
/** Take a statement with for loops marked for vectorization, and turn
* them into single statements that operate on vectors. The loops in
* question must have constant extent.
*/
Stmt vectorize_loops(const Stmt &s, const std::map<std::string, Function> &env, const Target &t);
} // namespace Internal
} // namespace Halide
#endif
Computing file changes ...
