[Relay][Transform] Safe check added for Merge Composite (apache#5562)

trevor-m · Jun 18, 2020 · bc3a53c · bc3a53c
1 parent 867716f
commit bc3a53c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/relay/transforms/merge_composite.cc b/src/relay/transforms/merge_composite.cc
@@ -113,6 +113,7 @@ class MergeCompositeWrapper : public ExprMutator {
   Expr ExtractPattern(const Call& pattern, const Call& root, Map<std::string, Array<Expr>>* var_map,
                       Map<Expr, Expr>* call_map) {
     // check to make sure both calls are to operators (not functions)
+    if (!root.defined()) return Expr();
     if (!pattern->op->IsInstance<OpNode>() || !root->op->IsInstance<OpNode>()) return Expr();
     if (pattern->op.as<OpNode>()->name != root->op.as<OpNode>()->name) return Expr();