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

Updated tests to align with manager parameters update

Browse source
This commit is contained in:
Alejandro Saucedo 2021-02-08 21:42:02 +00:00
parent aa25f980d6
commit d7fe53eae6
8 changed files with 11 additions and 49 deletions
Download patch file Download diff file Expand all files Collapse all files

14
test/TestManager.cpp
View file

@ -17,9 +17,6 @@ TEST(TestManager, EndToEndOpMultFlow)
mgr.rebuildTensors({ tensorOutput });
mgr.evalOpDefault<kp::OpTensorSyncDevice>(
{ tensorLHS, tensorRHS, tensorOutput });
mgr.evalOpDefault<kp::OpMult>({ tensorLHS, tensorRHS, tensorOutput });
mgr.evalOpDefault<kp::OpTensorSyncLocal>({ tensorOutput });
@ -46,10 +43,6 @@ TEST(TestManager, OpMultSequenceFlow)
sq->begin();
sq->record<kp::OpTensorSyncDevice>({ tensorLHS });
sq->record<kp::OpTensorSyncDevice>({ tensorRHS });
sq->record<kp::OpTensorSyncDevice>({ tensorOutput });
sq->record<kp::OpMult>({ tensorLHS, tensorRHS, tensorOutput });
sq->record<kp::OpTensorSyncLocal>({ tensorOutput });
@ -106,9 +99,6 @@ TEST(TestManager, TestMultipleTensorsAtOnce)
sq->begin();
sq->record<kp::OpTensorSyncDevice>(
{ tensorLHS, tensorRHS, tensorOutput });
sq->record<kp::OpMult>({ tensorLHS, tensorRHS, tensorOutput });
sq->record<kp::OpTensorSyncLocal>({ tensorOutput });
@ -127,10 +117,6 @@ TEST(TestManager, TestCreateInitTensor)
std::shared_ptr<kp::Tensor> tensorA = mgr.buildTensor({ 0, 1, 2 });
std::shared_ptr<kp::Tensor> tensorB = mgr.buildTensor({ 0, 0, 0 });
mgr.rebuildTensors({ tensorA, tensorB });
mgr.evalOpDefault<kp::OpTensorSyncDevice>({ tensorA, tensorB });
mgr.evalOpDefault<kp::OpTensorCopy>({ tensorA, tensorB });
mgr.evalOpDefault<kp::OpTensorSyncLocal>({ tensorB });