thek0tyara/llama-cpp-turboquant
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 4

Added test for data types

Browse source 
Signed-off-by: Alejandro Saucedo <axsauze@gmail.com>
This commit is contained in:
Alejandro Saucedo 2021-09-11 17:54:03 +01:00
parent 7db6f8c46f
commit 89fce2057e
1 changed files with 30 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

30
test/TestTensor.cpp
View file

@ -10,5 +10,35 @@ TEST(TestTensor, ConstructorData)
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->dataTypeMemorySize(), sizeof(float));
EXPECT_EQ(tensor->vector(), vec);
}
TEST(TestTensor, DataTypes)
{
kp::Manager mgr;
{
std::vector<float> vec{ 0, 1, 2 };
std::shared_ptr<kp::TensorT<float>> tensor = mgr.tensor(vec);
EXPECT_EQ(tensor->dataType(), kp::Tensor::TensorDataTypes::eFloat);
}
{
std::vector<int32_t> vec{ 0, 1, 2 };
std::shared_ptr<kp::TensorT<int32_t>> tensor = mgr.tensorT(vec);
EXPECT_EQ(tensor->dataType(), kp::Tensor::TensorDataTypes::eInt);
}
{
std::vector<uint32_t> vec{ 0, 1, 2 };
std::shared_ptr<kp::TensorT<uint32_t>> tensor = mgr.tensorT(vec);
EXPECT_EQ(tensor->dataType(), kp::Tensor::TensorDataTypes::eUnsignedInt);
}
{
std::vector<double> vec{ 0, 1, 2 };
std::shared_ptr<kp::TensorT<double>> tensor = mgr.tensorT(vec);
EXPECT_EQ(tensor->dataType(), kp::Tensor::TensorDataTypes::eDouble);
}
}