llama-cpp-turboquant/test/TestTensor.cpp
2021-02-27 14:49:13 +00:00

14 lines
295 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::Tensor> tensor = mgr.tensor(vec);
EXPECT_EQ(tensor->size(), vec.size());
EXPECT_EQ(tensor->data(), vec);
}