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
|
#include "caffe2/core/logging.h"
#include "caffe2/opt/converter.h"
#include "caffe2/opt/passes.h"
namespace caffe2 {
namespace opt {
using namespace nom;
using namespace nom::repr;
void deadCodeElim(NNModule* nn) {
// Iteratively remove unconsumed non-external outputs.
bool changed = false;
do {
changed = false;
for (const auto& node : nn->dataFlow.getMutableNodes()) {
NOM_REQUIRE_OR_CONT(nn::is<repr::NeuralNetOperator>(node));
bool isUsed = false;
for (const auto& output : nn::getOutputs(node)) {
if (nn::hasConsumer(output) || nn->outputs.count(output)) {
isUsed = true;
break;
}
}
NOM_REQUIRE_OR_CONT(!isUsed);
// No outputs are used, delete them and the node itself.
for (const auto& output : nn::getOutputs(node)) {
nn->dataFlow.deleteNode(output);
}
nn->dataFlow.deleteNode(node);
changed = true;
break;
}
} while (changed);
}
REGISTER_OPT_PASS_FROM_FUNC(DeadCodeElim, deadCodeElim);
} // namespace opt
} // namespace caffe2
|
