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

Updated tests to use shaderc

Browse source
This commit is contained in:
Alejandro Saucedo 2021-02-18 22:29:15 +00:00
parent 940a60958c
commit 5bc2d0dae7
5 changed files with 42 additions and 49 deletions
Download patch file Download diff file Expand all files Collapse all files

28
test/TestMultipleAlgoExecutions.cpp
View file

@ -3,6 +3,8 @@
#include "kompute/Kompute.hpp"
#include "TestUtils.cpp"
TEST(TestMultipleAlgoExecutions, SingleSequenceRecord)
{
@ -28,11 +30,11 @@ TEST(TestMultipleAlgoExecutions, SingleSequenceRecord)
sq->begin();
sq->record<kp::OpAlgoBase>(
{ tensorA }, std::vector<char>(shader.begin(), shader.end()));
{ tensorA }, spirv_from_string(shader));
sq->record<kp::OpAlgoBase>(
{ tensorA }, std::vector<char>(shader.begin(), shader.end()));
{ tensorA }, spirv_from_string(shader));
sq->record<kp::OpAlgoBase>(
{ tensorA }, std::vector<char>(shader.begin(), shader.end()));
{ tensorA }, spirv_from_string(shader));
sq->record<kp::OpTensorSyncLocal>({ tensorA });
@ -73,19 +75,19 @@ TEST(TestMultipleAlgoExecutions, MultipleCmdBufRecords)
// Then perform the computations
sq->begin();
sq->record<kp::OpAlgoBase>({ tensorA },
std::vector<char>(shader.begin(), shader.end()));
spirv_from_string(shader));
sq->end();
sq->eval();
sq->begin();
sq->record<kp::OpAlgoBase>({ tensorA },
std::vector<char>(shader.begin(), shader.end()));
spirv_from_string(shader));
sq->end();
sq->eval();
sq->begin();
sq->record<kp::OpAlgoBase>({ tensorA },
std::vector<char>(shader.begin(), shader.end()));
spirv_from_string(shader));
sq->end();
sq->eval();
@ -122,7 +124,7 @@ TEST(TestMultipleAlgoExecutions, MultipleSequences)
sq->begin();
sq->record<kp::OpAlgoBase>(
{ tensorA }, std::vector<char>(shader.begin(), shader.end()));
{ tensorA }, spirv_from_string(shader));
sq->end();
sq->eval();
@ -135,7 +137,7 @@ TEST(TestMultipleAlgoExecutions, MultipleSequences)
sq->begin();
sq->record<kp::OpAlgoBase>(
{ tensorA }, std::vector<char>(shader.begin(), shader.end()));
{ tensorA }, spirv_from_string(shader));
sq->end();
sq->eval();
@ -148,7 +150,7 @@ TEST(TestMultipleAlgoExecutions, MultipleSequences)
sq->begin();
sq->record<kp::OpAlgoBase>(
{ tensorA }, std::vector<char>(shader.begin(), shader.end()));
{ tensorA }, spirv_from_string(shader));
sq->end();
sq->eval();
@ -206,7 +208,7 @@ TEST(TestMultipleAlgoExecutions, SingleRecordMultipleEval)
sq->begin();
sq->record<kp::OpAlgoBase>(
{ tensorA }, std::vector<char>(shader.begin(), shader.end()));
{ tensorA }, spirv_from_string(shader));
sq->end();
@ -265,7 +267,7 @@ TEST(TestMultipleAlgoExecutions, ManagerEvalMultSourceStrOpCreate)
mgr.evalOpDefault<kp::OpAlgoBase>(
{ tensorInA, tensorInB, tensorOut },
std::vector<char>(shader.begin(), shader.end()));
spirv_from_string(shader));
mgr.evalOpDefault<kp::OpTensorSyncLocal>({ tensorOut });
@ -308,7 +310,7 @@ TEST(TestMultipleAlgoExecutions, ManagerEvalMultSourceStrMgrCreate)
mgr.evalOpDefault<kp::OpAlgoBase>(
{ tensorInA, tensorInB, tensorOut },
std::vector<char>(shader.begin(), shader.end()));
spirv_from_string(shader));
mgr.evalOpDefault<kp::OpTensorSyncLocal>({ tensorOut });
@ -340,7 +342,7 @@ TEST(TestMultipleAlgoExecutions, SequenceAlgoDestroyOutsideManagerScope)
sq->begin();
sq->record<kp::OpAlgoBase>(
{ tensorA }, std::vector<char>(shader.begin(), shader.end()));
{ tensorA }, spirv_from_string(shader));
sq->end();
sq->eval();