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

Updated todo comments

Browse source
This commit is contained in:
Alejandro Saucedo 2020-09-06 12:59:35 +01:00
parent 1b4e4b6b18
commit e6f4097acb
3 changed files with 29 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

29
test/TestOpTensorCopy.cpp
View file

@ -27,6 +27,35 @@ TEST(TestOpTensorCopy, CopyDeviceToDeviceTensor) {
EXPECT_EQ(tensorA->data(), tensorB->data());
}
TEST(TestOpTensorCopy, CopyDeviceToDeviceTensorMulti) {
kp::Manager mgr;
std::vector<float> testVecA{ 9, 8, 7 };
std::vector<float> testVecB{ 0, 0, 0 };
std::vector<float> testVecC{ 0, 0, 0 };
std::shared_ptr<kp::Tensor> tensorA{new kp::Tensor(testVecA)};
std::shared_ptr<kp::Tensor> tensorB{new kp::Tensor(testVecB)};
std::shared_ptr<kp::Tensor> tensorC{new kp::Tensor(testVecC)};
mgr.evalOpDefault<kp::OpTensorCreate>({tensorA, tensorB, tensorC});
EXPECT_TRUE(tensorA->isInit());
EXPECT_TRUE(tensorB->isInit());
EXPECT_TRUE(tensorC->isInit());
mgr.evalOpDefault<kp::OpTensorCopy>({tensorA, tensorB, tensorC});
EXPECT_EQ(tensorA->data(), tensorB->data());
EXPECT_EQ(tensorA->data(), tensorC->data());
// Making sure the GPU holds the same data
mgr.evalOpDefault<kp::OpTensorSyncLocal>({tensorB, tensorC});
EXPECT_EQ(tensorA->data(), tensorB->data());
EXPECT_EQ(tensorA->data(), tensorC->data());
}
TEST(TestOpTensorCopy, CopyDeviceToStagingTensor) {
kp::Manager mgr;