Skip to content

Commit 6dd54c0

Browse files
committed
fix typo
1 parent b31cacf commit 6dd54c0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

core/partitioning/partitioning.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@ void setNonTensorConnectedNodes(PartitioningCtx* ctx, std::vector<torch::jit::No
118118
if (!isTensor(output)) {
119119
for (auto use : output->uses()) {
120120
auto node = use.user;
121-
if (isConstantOrUninitialized(node) && ctx->shouldNodeRunInTensorRT(node)) {
121+
if (!isConstantOrUninitialized(node) && ctx->shouldNodeRunInTensorRT(node)) {
122122
ctx->setNodeExecutorDecision(node, NodeExecutorDecision::kNON_TENSOR);
123123
q.push(node);
124124
}

0 commit comments

Comments
 (0)