Function TEST

imperative/src/test/interpreter_test.cpp:20–47

using namespace interpreter;

TEST(TestImperative, InterpreterPut) {
    HostTensorGenerator<> gen;
    auto h0 = gen({3});
    auto&& channel = Interpreter::inst().create_channel();
    auto tensor_handle = channel->put(*h0, true);
    auto tensor_info = reinterpret_cast<intl::TensorInfo*>(tensor_handle);
    channel->sync();
    ASSERT_TRUE(tensor_info->status == intl::TensorInfo::Produced);
    //! because tensor elems is less than TensorShape::MAX_NDIM, stored it
    //! directly
    ASSERT_EQ(tensor_info->ptr->get_value().raw_ptr(), h0->raw_ptr());
    auto shape = channel->get_shape(tensor_handle);
    ASSERT_TRUE(shape.ndim == 1);
    ASSERT_TRUE(shape.total_nr_elems() == 3);

    auto h2 = gen({10});
    auto tensor_handle2 = channel->put(*h2, false);
    auto tensor_handle2_once = channel->put(*h2, false);
    channel->sync();
    ASSERT_NE(tensor_handle2_once, tensor_handle2);
    auto finded = MultiCNConstTensorCache::inst().lookup(*h2);
    ASSERT_TRUE(finded.get());
    //! Device tensor ptr is not equal host tensor ptr
    ASSERT_NE(finded->raw_ptr_not_for_readwrite(), h2->raw_ptr());
    channel->del(tensor_handle);
    channel->del(tensor_handle2);
    channel->del(tensor_handle2_once);
}

TEST(TestImperative, InterpreterApplyOp) {
    HostTensorGenerator<> gen;