Halide
AddAtomicMutex.h
Go to the documentation of this file.
1
#ifndef HALIDE_ADD_ATOMIC_MUTEX_H
2
#define HALIDE_ADD_ATOMIC_MUTEX_H
3
4
#include "
Expr.h
"
5
#include <map>
6
7
/** \file
8
* Defines the lowering pass that insert mutex allocation code & locks
9
* for the atomic nodes that require mutex locks. It also checks whether
10
* the atomic operation is valid. It rejects algorithms that have indexing
11
* on left-hand-side which references the buffer itself, e.g.
12
* f(clamp(f(r), 0, 100)) = f(r) + 1
13
* If the SplitTuple pass does not lift out the Provide value as a let
14
* expression. This is confirmed by checking whether the Provide nodes
15
* inside an Atomic node have let binding values accessing the buffers
16
* inside the atomic node.
17
* Finally, it lifts the store indexing expressions inside the atomic node
18
* outside of the atomic to avoid side-effects inside those expressions
19
* being evaluated twice. */
20
21
namespace
Halide
{
22
namespace
Internal
{
23
24
class
Function
;
25
26
Stmt
add_atomic_mutex
(Stmt s,
const
std::map<std::string, Function> &env);
27
28
}
// namespace Internal
29
}
// namespace Halide
30
31
#endif
Halide::Internal::ArgInfoKind::Function
@ Function
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::add_atomic_mutex
Stmt add_atomic_mutex(Stmt s, const std::map< std::string, Function > &env)
Expr.h
src
AddAtomicMutex.h
Generated by
1.8.17