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

Added initial advanced example of logistic regression implementation

Browse source
This commit is contained in:
Alejandro Saucedo 2020-08-31 22:48:27 +01:00
parent a325486d01
commit 8eb9853dc8
5 changed files with 244 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

21
test/TestLogisticRegression.cpp
View file

@ -33,20 +33,26 @@ TEST_CASE("test_logistic_regression") {
kp::Manager mgr;
if (std::shared_ptr<kp::Sequence> sq =
mgr.getOrCreateManagedSequence("testSq").lock()) {
mgr.getOrCreateManagedSequence("createTensors").lock()) {
sq->begin();
sq->record<kp::OpCreateTensor>(params);
sq->record<kp::OpAlgoBase<>>(
params,
true, // Whether to copy output from device
"test/shaders/glsl/test_logistic_regression.comp");
sq->end();
sq->eval();
// Iterate across all expected iterations
for (size_t i = 0; i < ITERATIONS; i++) {
sq->begin();
sq->record<kp::OpAlgoBase<>>(
params,
true, // Whether to copy output from device
"test/shaders/glsl/test_logistic_regression.comp");
sq->end();
sq->eval();
// TODO: Reference of data instead of full value copy every time
@ -75,4 +81,7 @@ TEST_CASE("test_logistic_regression") {
REQUIRE(wIn->data()[0] < 0.01);
REQUIRE(wIn->data()[1] > 1.0);
REQUIRE(bIn->data()[0] < 0.0);
SPDLOG_DEBUG("Result wIn: {}, bIn: {}",
wIn->data(), bIn->data());
}