Skip to content

Commit

Permalink
fix problem
Browse files Browse the repository at this point in the history
  • Loading branch information
electriclilies committed Nov 3, 2021
1 parent 35f2636 commit 831879c
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions src/relay/backend/te_compiler.cc
Original file line number Diff line number Diff line change
Expand Up @@ -593,8 +593,9 @@ class LowerTensorExprMutator : public DeviceAwareExprMutator {
}

Expr DeviceAwareVisitExpr_(const CallNode* call_node) override {
ICHECK(call_node->op != CallLoweredOp())
<< "Found a call_lowered op: can't lower a function that is already lowered.";
// Passes before lowering might insert a call_lowered to call a function that has already
// been lowered. Therefore we might see call_lowered ops here, but we don't need to do anything
// because ResolveToPrimitive returns null for all calls where the call_node->op is an OpNode
Call call = GetRef<Call>(call_node);
// Look for (indirect) calls to primitives.
BaseFunc prim_func = ResolveToPrimitive(call_node->op);
Expand Down

0 comments on commit 831879c

Please sign in to comment.