Halide
Memoization.h
Go to the documentation of this file.
1 #ifndef HALIDE_INTERNAL_CACHING_H
2 #define HALIDE_INTERNAL_CACHING_H
3 
4 /** \file
5  *
6  * Defines the interface to the pass that injects support for
7  * compute_cached roots.
8  */
9 
10 #include <map>
11 #include <string>
12 
13 #include "Expr.h"
14 
15 namespace Halide {
16 namespace Internal {
17 
18 class Function;
19 
20 /** Transform pipeline calls for Funcs scheduled with memoize to do a
21  * lookup call to the runtime cache implementation, and if there is a
22  * miss, compute the results and call the runtime to store it back to
23  * the cache.
24  * Should leave non-memoized Funcs unchanged.
25  */
26 Stmt inject_memoization(const Stmt &s, const std::map<std::string, Function> &env,
27  const std::string &name,
28  const std::vector<Function> &outputs);
29 
30 /** This should be called after Storage Flattening has added Allocation
31  * IR nodes. It connects the memoization cache lookups to the Allocations
32  * so they point to the buffers from the memoization cache and those buffers
33  * are released when no longer used.
34  * Should not affect allocations for non-memoized Funcs.
35  */
36 Stmt rewrite_memoized_allocations(const Stmt &s, const std::map<std::string, Function> &env);
37 
38 } // namespace Internal
39 } // namespace Halide
40 
41 #endif
Halide
This file defines the class FunctionDAG, which is our representation of a Halide pipeline,...
Definition: AddAtomicMutex.h:21
Halide::LinkageType::Internal
@ Internal
Not visible externally, similar to 'static' linkage in C.
Halide::Internal::IOKind::Function
@ Function
Expr.h
Halide::Internal::inject_memoization
Stmt inject_memoization(const Stmt &s, const std::map< std::string, Function > &env, const std::string &name, const std::vector< Function > &outputs)
Transform pipeline calls for Funcs scheduled with memoize to do a lookup call to the runtime cache im...
Halide::Internal::rewrite_memoized_allocations
Stmt rewrite_memoized_allocations(const Stmt &s, const std::map< std::string, Function > &env)
This should be called after Storage Flattening has added Allocation IR nodes.