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

Added TensorOpTensorCreate tests

Browse source
This commit is contained in:
Alejandro Saucedo 2021-02-27 07:42:38 +00:00
parent fd0f0d3eb4
commit 7fb1515c93
2 changed files with 54 additions and 1695 deletions
Download patch file Download diff file Expand all files Collapse all files

205
test/TestOpTensorCreate.cpp
View file

@ -3,155 +3,56 @@
#include "kompute/Kompute.hpp"
//TEST(TestOpTensorCreate, CreateSingleTensorSingleOp)
//{
// std::vector<float> testVecA{ 9, 8, 7 };
// std::shared_ptr<kp::Tensor> tensorA{ new kp::Tensor(testVecA) };
//
// {
// kp::Manager mgr;
//
// mgr.rebuild({ tensorA });
//
// EXPECT_TRUE(tensorA->isInit());
//
// EXPECT_EQ(tensorA->data(), testVecA);
// }
//
// EXPECT_FALSE(tensorA->isInit());
//}
//
//TEST(TestOpTensorCreate, CreateMultipleTensorSingleOp)
//{
//
// kp::Manager mgr;
//
// std::vector<float> testVecA{ 9, 8, 7 };
// std::vector<float> testVecB{ 6, 5, 4 };
//
// std::shared_ptr<kp::Tensor> tensorA{ new kp::Tensor(testVecA) };
// std::shared_ptr<kp::Tensor> tensorB{ new kp::Tensor(testVecB) };
//
// mgr.rebuild({ tensorA, tensorB });
//
// EXPECT_TRUE(tensorA->isInit());
// EXPECT_TRUE(tensorB->isInit());
//
// EXPECT_EQ(tensorA->data(), testVecA);
// EXPECT_EQ(tensorB->data(), testVecB);
//}
//
//TEST(TestOpTensorCreate, CreateMultipleTensorMultipleOp)
//{
//
// kp::Manager mgr;
//
// std::vector<float> testVecA{ 9, 8, 7 };
// std::vector<float> testVecB{ 6, 5, 4 };
//
// std::shared_ptr<kp::Tensor> tensorA{ new kp::Tensor(testVecA) };
// std::shared_ptr<kp::Tensor> tensorB{ new kp::Tensor(testVecB) };
//
// mgr.rebuild({ tensorA });
// mgr.rebuild({ tensorB });
//
// EXPECT_TRUE(tensorA->isInit());
// EXPECT_TRUE(tensorB->isInit());
//
// EXPECT_EQ(tensorA->data(), testVecA);
// EXPECT_EQ(tensorB->data(), testVecB);
//}
//
//TEST(TestOpTensorCreate, TestTensorMemoryManagedByManagerDestroyed)
//{
//
// std::vector<float> testVecA{ 9, 8, 7 };
// std::vector<float> testVecB{ 6, 5, 4 };
//
// std::shared_ptr<kp::Tensor> tensorA{ new kp::Tensor(testVecA) };
// std::shared_ptr<kp::Tensor> tensorB{ new kp::Tensor(testVecB) };
//
// {
// kp::Manager mgr;
// mgr.rebuild({ tensorA });
// mgr.rebuild({ tensorB });
//
// EXPECT_TRUE(tensorA->isInit());
// EXPECT_TRUE(tensorB->isInit());
//
// EXPECT_EQ(tensorA->data(), testVecA);
// EXPECT_EQ(tensorB->data(), testVecB);
// }
//
// EXPECT_FALSE(tensorA->isInit());
// EXPECT_FALSE(tensorB->isInit());
//}
//
//TEST(TestOpTensorCreate, TestTensorMemoryManagedByManagerNOTDestroyed)
//{
//
// std::vector<float> testVecA{ 9, 8, 7 };
// std::vector<float> testVecB{ 6, 5, 4 };
//
// std::shared_ptr<kp::Tensor> tensorA{ new kp::Tensor(testVecA) };
// std::shared_ptr<kp::Tensor> tensorB{ new kp::Tensor(testVecB) };
//
// kp::Manager mgr;
//
// {
// mgr.rebuild({ tensorA });
// mgr.rebuild({ tensorB });
//
// EXPECT_TRUE(tensorA->isInit());
// EXPECT_TRUE(tensorB->isInit());
//
// EXPECT_EQ(tensorA->data(), testVecA);
// EXPECT_EQ(tensorB->data(), testVecB);
// }
//
// EXPECT_TRUE(tensorA->isInit());
// EXPECT_TRUE(tensorB->isInit());
//}
//
//TEST(TestOpTensorCreate, NoErrorIfTensorFreedBefore)
//{
//
// std::vector<float> testVecA{ 9, 8, 7 };
// std::vector<float> testVecB{ 6, 5, 4 };
//
// std::shared_ptr<kp::Tensor> tensorA{ new kp::Tensor(testVecA) };
// std::shared_ptr<kp::Tensor> tensorB{ new kp::Tensor(testVecB) };
//
// kp::Manager mgr;
//
// mgr.rebuild({ tensorA });
// mgr.rebuild({ tensorB });
//
// EXPECT_TRUE(tensorA->isInit());
// EXPECT_TRUE(tensorB->isInit());
//
// EXPECT_EQ(tensorA->data(), testVecA);
// EXPECT_EQ(tensorB->data(), testVecB);
//
// tensorA->freeMemoryDestroyGPUResources();
// tensorB->freeMemoryDestroyGPUResources();
// EXPECT_FALSE(tensorA->isInit());
// EXPECT_FALSE(tensorB->isInit());
//}
//
//TEST(TestOpTensorCreate, ExceptionOnZeroSizeTensor)
//{
// std::vector<float> testVecA;
//
// std::shared_ptr<kp::Tensor> tensorA{ new kp::Tensor(testVecA) };
//
// kp::Manager mgr;
//
// try {
// mgr.rebuild({ tensorA });
// } catch (const std::runtime_error& err) {
// // check exception
// ASSERT_TRUE(std::string(err.what()).find("zero-sized") !=
// std::string::npos);
// }
//}
TEST(TestOpTensorCreate, CreateSingleTensorSingleOp)
{
std::vector<float> testVecA{ 9, 8, 7 };
std::shared_ptr<kp::Tensor> tensorA = nullptr;
{
kp::Manager mgr;
tensorA = mgr.tensor(testVecA);
EXPECT_TRUE(tensorA->isInit());
EXPECT_EQ(tensorA->data(), testVecA);
}
EXPECT_FALSE(tensorA->isInit());
}
TEST(TestOpTensorCreate, NoErrorIfTensorFreedBefore)
{
std::vector<float> testVecA{ 9, 8, 7 };
std::vector<float> testVecB{ 6, 5, 4 };
kp::Manager mgr;
std::shared_ptr<kp::Tensor> tensorA = mgr.tensor(testVecA);
std::shared_ptr<kp::Tensor> tensorB = mgr.tensor(testVecB);
EXPECT_EQ(tensorA->data(), testVecA);
EXPECT_EQ(tensorB->data(), testVecB);
tensorA->destroy();
tensorB->destroy();
EXPECT_FALSE(tensorA->isInit());
EXPECT_FALSE(tensorB->isInit());
}
TEST(TestOpTensorCreate, ExceptionOnZeroSizeTensor)
{
std::vector<float> testVecA;
kp::Manager mgr;
try {
std::shared_ptr<kp::Tensor> tensorA = mgr.tensor(testVecA);
} catch (const std::runtime_error& err) {
// check exception
ASSERT_TRUE(std::string(err.what()).find("zero-sized") !=
std::string::npos);
}
}