forked from marian-nmt/marian-dev
/
node.cpp
executable file
·90 lines (76 loc) · 1.75 KB
/
node.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
#include "graph/node.h"
#include "graph/auto_tuner.h"
#include "graph/expression_graph.h"
#include "tensors/backend.h"
namespace marian {
void Node::allocate() {
if(!val_) {
graph()->allocateForward(this);
}
}
void Node::free() {
if(destroy_) { // don't free views, @TODO: better naming
//std::cerr << "Freeing" << std::endl;
if(graph()) {
if(val_) {
graph()->free(val_);
val_ = nullptr;
}
if(adj_) {
graph()->free(adj_);
adj_ = nullptr;
}
}
}
}
/**
* Initialization for backward step of top node
* in computation graph. Allocates memory and sets gradient
* to 1 (df/df == 1).
*/
void Node::init_dependent() {
if(!adj_) {
graph()->allocateBackward(this);
adj_->set(1.f);
}
}
/**
* Initialization for backward step of any non-top node
* in computation graph. Allocates memory and sets gradient
* to 0 for further accumulation of gradients from all
* parents.
*/
void Node::set_zero_adjoint() {
if(!adj_) {
graph()->allocateBackward(this);
adj_->set(0.f);
}
}
float Node::scalar() {
return val_->scalar();
}
Ptr<Backend> Node::getBackend() {
return graph()->getBackend();
}
void Node::forward() {
if(recorder_)
recorder_->start(recorderHash_);
runForward(forwardOps());
if(recorder_)
recorder_->stop(recorderHash_, recorderStop_);
}
void Node::backward() {
if(recorder_)
recorder_->start(recorderHash_);
runBackward(backwardOps());
if(recorder_ && recorderStop_)
recorder_->stop(recorderHash_, recorderStop_);
}
void Node::record(Ptr<AutoTunerRecorder> recorder,
size_t recorderHash,
bool stop) {
recorder_ = recorder;
recorderHash_ = recorderHash;
recorderStop_ = stop;
}
} // namespace marian