Function TEST

src/core/test/static_infer.cpp:120–145

}  // anonymous namespace

TEST(TestStaticInfer, ValueInfer) {
    using namespace cg::static_infer;
    HostTensorGenerator<> gen;
    constexpr size_t SIZE = 3;
    auto host_x0 = gen({SIZE}), host_x1 = gen({SIZE});

    auto graph = ComputingGraph::make();
    auto x0 = opr::Host2DeviceCopy::make(*graph, host_x0),
         x1 = opr::Host2DeviceCopy::make(*graph, host_x1), x2 = x0 + x1,
         y0 = x0.make_scalar(2.f), y1 = x0.make_scalar(3.f), y2 = opr::pow(y0, y1);
    auto&& mgr = x0.node()->owner_graph()->static_infer_manager();
    ASSERT_EQ(InferType::RT_STATIC, mgr.get_infer_type(x2.node()).value);
    ASSERT_EQ(InferType::CONST, mgr.get_infer_type(y0.node()).value);
    ASSERT_EQ(InferType::CONST, mgr.get_infer_type(y1.node()).value);
    ASSERT_EQ(InferType::CONST, mgr.get_infer_type(y2.node()).value);
    auto x2v = mgr.infer_value(x2.node());
    ASSERT_EQ(host_x0->shape(), x2v.shape());
    for (size_t i = 0; i < SIZE; i++)
        MGB_ASSERT_FLOAT_EQ(
                host_x0->ptr<float>()[i] + host_x1->ptr<float>()[i],
                x2v.ptr<float>()[i]);

    auto y2v = mgr.infer_value(y2.node());
    ASSERT_TRUE(y2v.shape().is_scalar());
    MGB_ASSERT_FLOAT_EQ(8.f, y2v.ptr<float>()[0]);
}

TEST(TestStaticInfer, ValueNonContig) {
    using namespace cg::static_infer;