Revision 5deee77b677da97c4dadc966f76d8bf01d855c57 authored by Steven Johnson on 01 April 2020, 00:47:39 UTC, committed by Steven Johnson on 01 April 2020, 00:47:39 UTC
1 parent cb6f4a9
Prefetch.h
#ifndef HALIDE_PREFETCH_H
#define HALIDE_PREFETCH_H
/** \file
* Defines the lowering pass that injects prefetch calls when prefetching
* appears in the schedule.
*/
#include <map>
#include <string>
#include <vector>
#include "Expr.h"
namespace Halide {
struct Target;
namespace Internal {
class Function;
struct PrefetchDirective;
/** Inject placeholder prefetches to 's'. This placholder prefetch
* does not have explicit region to be prefetched yet. It will be computed
* during call to \ref inject_prefetch. */
Stmt inject_placeholder_prefetch(const Stmt &s, const std::map<std::string, Function> &env,
const std::string &prefix,
const std::vector<PrefetchDirective> &prefetches);
/** Compute the actual region to be prefetched and place it to the
* placholder prefetch. Wrap the prefetch call with condition when
* applicable. */
Stmt inject_prefetch(const Stmt &s, const std::map<std::string, Function> &env);
/** Reduce a multi-dimensional prefetch into a prefetch of lower dimension
* (max dimension of the prefetch is specified by target architecture).
* This keeps the 'max_dim' innermost dimensions and adds loops for the rest
* of the dimensions. If maximum prefetched-byte-size is specified (depending
* on the architecture), this also adds an outer loops that tile the prefetches. */
Stmt reduce_prefetch_dimension(Stmt stmt, const Target &t);
} // namespace Internal
} // namespace Halide
#endif
![swh spinner](/static/img/swh-spinner.gif)
Computing file changes ...