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

Added fix for test end to end functionality

Browse source
This commit is contained in:
Alejandro Saucedo 2021-03-04 19:30:02 +00:00
parent 0558a1b6fc
commit ee4ba75bea
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
test/TestMultipleAlgoExecutions.cpp
View file

@ -49,12 +49,12 @@ TEST(TestMultipleAlgoExecutions, TestEndToEndFunctionality)
kp::Constants pushConstsB({ 3.0 });
auto algorithm = mgr.algorithm(
params, kp::Shader::compile_source(shader), workgroup, specConsts);
params, kp::Shader::compile_source(shader), workgroup, specConsts, pushConstsA);
// 3. Run operation with string shader synchronously
mgr.sequence()
->record<kp::OpTensorSyncDevice>(params)
->record<kp::OpAlgoDispatch>(algorithm, pushConstsA)
->record<kp::OpAlgoDispatch>(algorithm)
->record<kp::OpAlgoDispatch>(algorithm, pushConstsB)
->eval();