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

Added a buildTensor command to manager to simplify tensor creation workflows

Browse source
This commit is contained in:
Alejandro Saucedo 2020-09-10 08:35:27 +01:00
parent dcd8340b27
commit 4171786b6f
5 changed files with 147 additions and 77 deletions
Download patch file Download diff file Expand all files Collapse all files

20
test/TestManager.cpp
View file

@ -107,3 +107,23 @@ TEST(TestManager, TestMultipleTensorsAtOnce) {
EXPECT_EQ(tensorOutput->data(), std::vector<float>({0, 4, 12}));
}
TEST(TestManager, TestCreateInitTensor) {
kp::Manager mgr;
std::shared_ptr<kp::Tensor> tensorA = mgr.buildTensor({0,1,2});
std::shared_ptr<kp::Tensor> tensorB = mgr.buildTensor({0,0,0});
mgr.evalOpDefault<kp::OpTensorCopy>({tensorA, tensorB});
mgr.evalOpDefault<kp::OpTensorSyncLocal>({tensorB});
EXPECT_EQ(tensorB->data(), std::vector<float>({0,1,2}));
std::shared_ptr<kp::Tensor> tensorC = mgr.buildTensor({0,0,0}, kp::Tensor::TensorTypes::eStaging);
mgr.evalOpDefault<kp::OpTensorCopy>({tensorA, tensorC});
EXPECT_EQ(tensorC->data(), std::vector<float>({0,1,2}));
}