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

Added optensorSyncDevice and optensorcopy tests

Browse source
This commit is contained in:
Alejandro Saucedo 2020-09-06 12:52:45 +01:00
parent ec89fc6d56
commit 1b4e4b6b18
12 changed files with 398 additions and 116 deletions
Download patch file Download diff file Expand all files Collapse all files

4
test/TestTensor.cpp
View file

@ -18,7 +18,7 @@ TEST(TestTensor, CopyFromHostData) {
vecA,
kp::Tensor::TensorTypes::eStaging);
std::shared_ptr<kp::Tensor> tensorB = std::make_shared<kp::Tensor>(
vecA,
vecB,
kp::Tensor::TensorTypes::eStaging);
kp::Manager mgr;
@ -35,8 +35,6 @@ TEST(TestTensor, CopyFromHostData) {
sq->end();
sq->eval();
tensorB->mapDataFromHostMemory();
}
EXPECT_EQ(tensorA->data(), tensorB->data());