[BYOC][MergeComposite] if root->args[i] isn't a CallNode, then Donwca…

…st<Call> will check fail Signed-off-by: windclarion <windclarion@gmail.com>
apache · May 19, 2020 · 5bee3d9 · 5bee3d9
1 parent 538e296
commit 5bee3d9
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/relay/transforms/merge_composite.cc b/src/relay/transforms/merge_composite.cc
@@ -122,8 +122,10 @@ class MergeCompositeWrapper : public ExprMutator {
     for (const auto& arg : pattern->args) {
       Expr new_arg;
       if (arg->IsInstance<CallNode>()) {
-        new_arg =
-            ExtractPattern(Downcast<Call>(arg), Downcast<Call>(root->args[i]), var_map, call_map);
+        if (root->args[i]->IsInstance<CallNode>()) {
+          new_arg =
+              ExtractPattern(Downcast<Call>(arg), Downcast<Call>(root->args[i]), var_map, call_map);
+        }
         // if we've already processed this call node, return the previous result
         if (call_map->find(arg) != call_map->end() && new_arg.defined()) {
           new_arg = (*call_map)[arg];