Function TEST

src/opr/test/megdnn_wrapper.cpp:11–46

using namespace mgb;

TEST(TestOprMegDNNWrapper, Stream) {
    using Param = opr::Convolution::Param;
    Param param;
    HostTensorGenerator<> gen;
    auto host_x = gen({8, 1, 20, 20}), host_kern = gen({3, 1, 4, 4});
    HostTensorND host_y_expect;
    {
        // gen host_y_expect
        auto graph = ComputingGraph::make();
        auto x = opr::Host2DeviceCopy::make(*graph, host_x),
             kern = opr::Host2DeviceCopy::make(*graph, host_kern),
             y = opr::Convolution::make(x, kern, param);
        auto func = graph->compile({make_callback_copy(y, host_y_expect)});
        func->execute();
    }
    auto graph = ComputingGraph::make();
    auto x = opr::Host2DeviceCopy::make(*graph, host_x),
         kern = opr::Host2DeviceCopy::make(*graph, host_kern),
         y = opr::Convolution::make(x, kern, param);
    // change stream
    auto chg = [](SymbolVar var) {
        auto opr = var.node()->owner_opr();
        for (auto i : opr->output())
            i->comp_node(CompNode::load("xpu0:1"));
        opr->on_output_comp_node_stream_changed();
    };
    chg(x);
    chg(kern);
    chg(y);
    HostTensorND host_y;

    opr::Sleep::sleep(host_x->comp_node(), 0.5);
    auto func = graph->compile({make_callback_copy(y, host_y)});
    func->execute();
    MGB_ASSERT_TENSOR_EQ(host_y_expect, host_y);
}

TEST(TestOprMegDNNWrapper, ShapeDep) {
    using Param = opr::Convolution::Param;