Function TEST

src/plugin/test/cpu_dispatch_checker.cpp:9–20

using namespace mgb;

TEST(TestCPUDispatchChecker, Simple) {
    HostTensorGenerator<> gen;
    auto graph = ComputingGraph::make();
    graph->options().var_sanity_check_first_run = false;
    CPUDispatchChecker checker(graph.get());
    auto host_x = gen({3}, CompNode::load("cpux"));
    auto x = opr::Host2DeviceCopy::make(*graph, host_x),
         y = opr::CallbackInjector::make(x, [](DeviceTensorND&) {}), z = y + 1;
    auto func = graph->compile({{z, {}}});
    func->execute();
    ASSERT_EQ(1u, checker.failed_oprs().count(y.node()->owner_opr()));
}

TEST(TestCPUDispatchChecker, Loop) {
    HostTensorGenerator<> gen;