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

updated tests on constants and workgroup typedefs

Browse source
This commit is contained in:
Alejandro Saucedo 2021-02-14 07:30:05 +00:00
parent 9adfa34fd3
commit e481c3afac
2 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

10
test/TestLogisticRegression.cpp
View file

@ -44,7 +44,7 @@ TEST(TestLogisticRegressionAlgorithm, TestMainLogisticRegression)
#ifdef KOMPUTE_SHADER_FROM_STRING #ifdef KOMPUTE_SHADER_FROM_STRING
sq->record<kp::OpAlgoBase>( sq->record<kp::OpAlgoBase>(
params, "test/shaders/glsl/test_logistic_regression.comp", params, "test/shaders/glsl/test_logistic_regression.comp",
std::array<uint32_t, 3>(), std::vector<float>({5.0})); kp::Workgroup(), kp::Constants({5.0}));
#else #else
sq->record<kp::OpAlgoBase>( sq->record<kp::OpAlgoBase>(
params, params,
@ -52,7 +52,7 @@ TEST(TestLogisticRegressionAlgorithm, TestMainLogisticRegression)
kp::shader_data::shaders_glsl_logisticregression_comp_spv, kp::shader_data::shaders_glsl_logisticregression_comp_spv,
kp::shader_data::shaders_glsl_logisticregression_comp_spv + kp::shader_data::shaders_glsl_logisticregression_comp_spv +
kp::shader_data::shaders_glsl_logisticregression_comp_spv_len), kp::shader_data::shaders_glsl_logisticregression_comp_spv_len),
std::array<uint32_t, 3>(), std::vector<float>({5.0})); kp::Workgroup(), kp::Constants({5.0}));
#endif #endif
sq->record<kp::OpTensorSyncLocal>({ wOutI, wOutJ, bOut, lOut }); sq->record<kp::OpTensorSyncLocal>({ wOutI, wOutJ, bOut, lOut });
@ -93,7 +93,7 @@ TEST(TestLogisticRegressionAlgorithm, TestMainLogisticRegressionManualCopy)
uint32_t ITERATIONS = 100; uint32_t ITERATIONS = 100;
float learningRate = 0.1; float learningRate = 0.1;
std::vector<float> wInVec = { 0.001, 0.001 }; kp::Constants wInVec = { 0.001, 0.001 };
std::vector<float> bInVec = { 0 }; std::vector<float> bInVec = { 0 };
std::shared_ptr<kp::Tensor> xI{ new kp::Tensor({ 0, 1, 1, 1, 1 }) }; std::shared_ptr<kp::Tensor> xI{ new kp::Tensor({ 0, 1, 1, 1, 1 }) };
@ -129,7 +129,7 @@ TEST(TestLogisticRegressionAlgorithm, TestMainLogisticRegressionManualCopy)
#ifdef KOMPUTE_SHADER_FROM_STRING #ifdef KOMPUTE_SHADER_FROM_STRING
sq->record<kp::OpAlgoBase>( sq->record<kp::OpAlgoBase>(
params, "test/shaders/glsl/test_logistic_regression.comp.spv", params, "test/shaders/glsl/test_logistic_regression.comp.spv",
std::array<uint32_t, 3>(), kp::Algorithm::SpecializationContainer{{(uint32_t)5}}); kp::Workgroup(), kp::Algorithm::SpecializationContainer{{(uint32_t)5}});
#else #else
sq->record<kp::OpAlgoBase>( sq->record<kp::OpAlgoBase>(
params, params,
@ -137,7 +137,7 @@ TEST(TestLogisticRegressionAlgorithm, TestMainLogisticRegressionManualCopy)
kp::shader_data::shaders_glsl_logisticregression_comp_spv, kp::shader_data::shaders_glsl_logisticregression_comp_spv,
kp::shader_data::shaders_glsl_logisticregression_comp_spv + kp::shader_data::shaders_glsl_logisticregression_comp_spv +
kp::shader_data::shaders_glsl_logisticregression_comp_spv_len), kp::shader_data::shaders_glsl_logisticregression_comp_spv_len),
std::array<uint32_t, 3>(), std::vector<float>({5.0})); kp::Workgroup(), kp::Constants({5.0}));
#endif #endif
sq->record<kp::OpTensorSyncLocal>({ wOutI, wOutJ, bOut, lOut }); sq->record<kp::OpTensorSyncLocal>({ wOutI, wOutJ, bOut, lOut });

4
test/TestMultipleAlgoExecutions.cpp
View file

@ -379,13 +379,13 @@ TEST(TestMultipleAlgoExecutions, TestAlgorithmSpecialized)
sq = mgr.sequence(); sq = mgr.sequence();
auto spec = std::vector<float>({5.0, 0.3}); auto spec = kp::Constants({5.0, 0.3});
sq->begin(); sq->begin();
sq->record<kp::OpAlgoBase>( sq->record<kp::OpAlgoBase>(
{ tensorA, tensorB }, { tensorA, tensorB },
std::vector<char>(shader.begin(), shader.end()), std::vector<char>(shader.begin(), shader.end()),
std::array<uint32_t, 3>(), spec); kp::Workgroup(), spec);
sq->end(); sq->end();
sq->eval(); sq->eval();