Function TEST

src/plugin/test/var_sanity_check.cpp:10–47

using namespace mgb;

TEST(TestVarSanityCheck, Simple) {
    HostTensorGenerator<> gen;
    auto host_x = gen({1024}), host_y = gen({1024});
    auto graph = ComputingGraph::make();
    SymbolVar x = opr::Host2DeviceCopy::make(*graph, host_x),
              y = opr::Host2DeviceCopy::make(*graph, host_y), y1 = y.reshape({1024, 1}),
              z = x + y1.reshape({1024});

    bool should_change = false;
    ComputingGraph::OutputSpec out_spec = {
            {y1,
             [&](DeviceTensorND& v) {
                 if (should_change) {
                     HostTensorND hv;
                     hv.copy_from(v).sync().ptr<float>()[123]++;
                     v.copy_from(hv);
                 }
             }},
            {z, [&](DeviceTensorND& v) {
                 HostTensorND hv;
                 hv.copy_from(v).sync();
                 for (int i = 0; i < 1024; i++) {
                     ASSERT_EQ(
                             host_x->ptr<float>()[i] + host_y->ptr<float>()[i],
                             hv.ptr<float>()[i])
                             << "failed at " << i;
                     ;
                 }
             }}};
    auto func = graph->compile(out_spec);
    func->execute().wait();
    func = graph->compile(out_spec);
    func->execute().wait();

    should_change = true;
    func = graph->compile(out_spec);
    ASSERT_THROW(func->execute().wait(), VarSanityCheck::Error);
}

TEST(TestVarSanityCheck, InputModify) {
    HostTensorGenerator<> gen;