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

Updated logistic regression model

Browse source
This commit is contained in:
Alejandro Saucedo 2020-11-03 18:04:03 +00:00
parent 157b6592dd
commit 5822850ef2
3 changed files with 37 additions and 61 deletions
Download patch file Download diff file Expand all files Collapse all files

14
examples/logistic_regression/src/Main.cpp
View file

@ -36,22 +36,30 @@ int main()
kp::Manager mgr;
std::shared_ptr<kp::Sequence> sqTensor =
mgr.createManagedSequence().lock();
mgr.createManagedSequence();
sqTensor->begin();
sqTensor->record<kp::OpTensorCreate>(params);
sqTensor->end();
sqTensor->eval();
std::shared_ptr<kp::Sequence> sq = mgr.createManagedSequence().lock();
std::shared_ptr<kp::Sequence> sq = mgr.createManagedSequence();
// Record op algo base
sq->begin();
sq->record<kp::OpTensorSyncDevice>({ wIn, bIn });
sq->record<kp::OpAlgoBase<>>(
#ifdef KOMPUTE_ANDROID_SHADER_FROM_STRING
sq->record<kp::OpAlgoBase>(
params, "shaders/glsl/logistic_regression.comp");
#else
sq->record<kp::OpAlgoBase>(
params, std::vector<char>(
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));
#endif
sq->record<kp::OpTensorSyncLocal>({ wOutI, wOutJ, bOut, lOut });