OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

add GRAPH_UPDATE logging to guard_elimination.cpp

Browse Source 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/22497

Differential Revision: D16165106

Pulled By: Krovatkin

fbshipit-source-id: aeb48d81d92c71f7038903b1656d760b6b95c562
This commit is contained in:
Nikolay Korovaiko 2019-07-09 10:01:57 -07:00 committed by Facebook Github Bot
parent a3346e100e
commit 073fa6f411
1 changed files with 20 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
torch/csrc/jit/passes/guard_elimination.cpp
View File

@ -1,5 +1,6 @@
#include <torch/csrc/jit/passes/alias_analysis.h>
#include <torch/csrc/jit/passes/guard_elimination.h>
#include <torch/csrc/jit/jit_log.h>
#include <torch/csrc/jit/passes/alias_analysis.h>
#include <memory>
#include <unordered_set>
@ -13,8 +14,11 @@ struct GuardElimination {
void run() {
moveGuardsToDefs(graph_->block());
GRAPH_DUMP("After moveGuardsToDefs", graph_);
coalesceGuards(graph_->block());
GRAPH_DUMP("After coalesceGuards", graph_);
eliminateRedundantGuards(graph_->block());
GRAPH_DUMP("After eliminateRedundantGuards", graph_);
}
void moveGuardsToDefs(Block* b) {
@ -31,7 +35,14 @@ struct GuardElimination {
if (guardee->owningBlock() != n->owningBlock()) {
guardee = *n->owningBlock()->nodes().begin();
}
aliasDb_->moveAfterTopologicallyValid(n, guardee);
bool moved = aliasDb_->moveAfterTopologicallyValid(n, guardee);
if (moved) {
GRAPH_UPDATE(
"Moved ",
n->output()->debugName(),
" to ",
n->inputs().at(0)->debugName());
}
} else {
it++;
for (Block* ib : n->blocks()) {
@ -55,6 +66,11 @@ struct GuardElimination {
if (inputs_to_guards.count(n->input())) {
auto prev = inputs_to_guards[n->input()];
n->output()->replaceAllUsesWith(prev->output());
GRAPH_UPDATE(
"Replacing ",
n->output()->debugName(),
" with ",
prev->output()->debugName());
it.destroyCurrent();
} else {
inputs_to_guards.insert({n->input(), n});
@ -94,6 +110,8 @@ struct GuardElimination {
auto pttp = n->output()->type();
n->output()->replaceAllUsesWith(n->inputs().at(0));
n->inputs().at(0)->setType(pttp);
GRAPH_UPDATE(
"Eliminating the redundant guard ", n->output()->debugName());
it.destroyCurrent();
} else {
it++;