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

Added manager util tests

Browse source 
Signed-off-by: Alejandro Saucedo <axsauze@gmail.com>
This commit is contained in:
Alejandro Saucedo 2021-09-01 07:06:56 +01:00
parent fab1bafd67
commit 8418d05455
1 changed files with 26 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

26
test/TestManager.cpp
View file

@ -78,3 +78,29 @@ TEST(TestManager, TestListDevices)
EXPECT_GT(devices.size(), 0);
EXPECT_GT(devices[0].getProperties().deviceName.size(), 0);
}
TEST(TestManager, TestClearDestroy)
{
kp::Manager mgr;
// Running within scope to run clear
{
std::shared_ptr<kp::TensorT<float>> tensorLHS = mgr.tensor({ 0, 1, 2 });
std::shared_ptr<kp::TensorT<float>> tensorRHS = mgr.tensor({ 2, 4, 6 });
std::shared_ptr<kp::TensorT<float>> tensorOutput = mgr.tensor({ 0, 0, 0 });
std::vector<std::shared_ptr<kp::Tensor>> params = { tensorLHS,
tensorRHS,
tensorOutput };
mgr.sequence()->eval<kp::OpTensorSyncDevice>(params);
mgr.sequence()->eval<kp::OpMult>(params, mgr.algorithm());
mgr.sequence()->eval<kp::OpTensorSyncLocal>(params);
EXPECT_EQ(tensorOutput->vector(), std::vector<float>({ 0, 4, 12 }));
}
mgr.clear();
mgr.destroy();
}