defensive program

pytorch · Jan 29, 2021 · 1cc7997 · 1cc7997
1 parent 64fde2a
commit 1cc7997
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torch/csrc/jit/passes/onnx/remove_inplace_ops_for_onnx.cpp b/torch/csrc/jit/passes/onnx/remove_inplace_ops_for_onnx.cpp
@@ -771,6 +771,7 @@ static void PrepareForRemoveMutations(MutationRemover& mr, Block* b) {
           b->prependNode(newNode);
           noneNode->insertBefore(newNode);
         }
+        TORCH_INTERNAL_ASSERT(nullptr != newNode);
         node->replaceInput(index, newNode->output());
         input->replaceAllUsesAfterNodeWith(node, newNode->output());
       }