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

Updated array example

Browse source
This commit is contained in:
Alejandro Saucedo 2020-11-03 08:42:27 +00:00
parent 0d33dc018d
commit b636a80d06
4 changed files with 36 additions and 37 deletions
Download patch file Download diff file Expand all files Collapse all files

11
examples/array_multiplication/src/Main.cpp
View file

@ -18,6 +18,7 @@ int main()
auto tensorInB = mgr.buildTensor({ 0.0, 1.0, 2.0 });
auto tensorOut = mgr.buildTensor({ 0.0, 0.0, 0.0 });
#ifdef KOMPUTE_ANDROID_SHADER_FROM_STRING
std::string shader(R"(
// The version to use
#version 450
@ -37,9 +38,17 @@ int main()
}
)");
mgr.evalOpDefault<kp::OpAlgoBase<>>(
mgr.evalOpDefault<kp::OpAlgoBase>(
{ tensorInA, tensorInB, tensorOut },
std::vector<char>(shader.begin(), shader.end()));
#else
mgr.evalOpDefault<kp::OpAlgoBase>(
{ tensorInA, tensorInB, tensorOut },
std::vector<char>(
kp::shader_data::shaders_glsl_opmult_comp_spv,
kp::shader_data::shaders_glsl_opmult_comp_spv
+ kp::shader_data::shaders_glsl_opmult_comp_spv_len));
#endif
mgr.evalOpDefault<kp::OpTensorSyncLocal>({tensorOut});