Skip to content

Commit 8f27e94

Browse files
committed
chore: add logging if found a modifying node in partitioning
Signed-off-by: Bo Wang <[email protected]>
1 parent 7af7169 commit 8f27e94

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

core/partitioning/partitioning.cpp

+1
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@ bool isModifyingNodes(torch::jit::Node* node, torch::jit::Value* val) {
3939
if (node->inputs()[i] == val) {
4040
const at::AliasInfo* formal = schema->arguments()[i].alias_info();
4141
if (formal && formal->isWrite()) {
42+
LOG_GRAPH("<Whatever is doing the modifying> Is modifying node " << util::node_info(n));
4243
return true;
4344
}
4445
}

0 commit comments

Comments
 (0)