llama-cpp-turboquant/test/TestTensor.cpp
2021-03-07 14:08:57 +00:00

13 lines
304 B
C++

#include "gtest/gtest.h"
#include "kompute/Kompute.hpp"
TEST(TestTensor, ConstructorData)
{
kp::Manager mgr;
std::vector<float> vec{ 0, 1, 2 };
std::shared_ptr<kp::TensorT<float>> tensor = mgr.tensor(vec);
EXPECT_EQ(tensor->size(), vec.size());
EXPECT_EQ(tensor->vector(), vec);
}