diff --git a/src/relay/transforms/annotate_target.cc b/src/relay/transforms/annotate_target.cc index c3d34cb9ab7c..454fa735db8e 100644 --- a/src/relay/transforms/annotate_target.cc +++ b/src/relay/transforms/annotate_target.cc @@ -48,8 +48,9 @@ class AnnotateTargetWrapper : public ExprMutator { bool IsSupported(const Expr& expr) { if (expr->IsInstance()) { Call call = Downcast(expr); - auto fannotate = Op::GetAttr("target." + target_); if (call->op->IsInstance()) { + if (!Op::HasAttr("target." + target_)) return false; + auto fannotate = Op::GetAttr("target." + target_); Op op = Downcast(call->op); CHECK(op.defined()); if (fannotate.count(op)) {