We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7af7169 commit 8f27e94Copy full SHA for 8f27e94
core/partitioning/partitioning.cpp
@@ -39,6 +39,7 @@ bool isModifyingNodes(torch::jit::Node* node, torch::jit::Value* val) {
39
if (node->inputs()[i] == val) {
40
const at::AliasInfo* formal = schema->arguments()[i].alias_info();
41
if (formal && formal->isWrite()) {
42
+ LOG_GRAPH("<Whatever is doing the modifying> Is modifying node " << util::node_info(n));
43
return true;
44
}
45
0 commit comments