Method apply

src/gopt/impl/misc.cpp:86–120

}

void ExpandVirtualGradPass::apply(OptState& opt) const {
    MIDOUT_B("ExpandVirtualGradPass::apply")
#if MGB_ENABLE_GRAD
    opt.set_var_replace_check_flag(VarReplaceCheckFlag::NOCHECK);
    auto rewriter = opt.graph().make_rewriter();
    auto on_opr = [&](OperatorNodeBase* opr) {
        if (!opr->same_type<opr::VirtualGrad>()) {
            rewriter.auto_replace_outputs(opr);
            return;
        }
        // Create opr and replace var but no need to copy old opr_properties
        // to new oprs because grad_manager would handle it.
        opt.call_with_opr(
                opr,
                [&] {
                    auto target = opr->input(0), wrt = opr->input(1),
                         grad = cg::grad(target, wrt).node();
                    auto src = opr->output(0);
                    grad = GraphOptimizer::var_replace_lookup(grad);
                    rewriter.replace_var(
                            src, grad,
                            mgb_ssprintf_log(
                                    "grad(%s, %s)", target->cname(), wrt->cname())
                                    .c_str());
                },
                OprPropertyFlag::NONE);
    };

    opt.graph().iter(on_opr);
    rewriter.apply_inplace();
#else
    MGB_MARK_USED_VAR(opt);
#endif
    MIDOUT_E
}

/* ================= DelayBroadcastPass ================ */