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

Added further tests for ci

Browse source
This commit is contained in:
Alejandro Saucedo 2021-01-20 21:21:07 +00:00
parent 661a50e62d
commit fbfef16105
3 changed files with 14 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

4
test/TestLogisticRegression.cpp
View file

@ -45,7 +45,7 @@ TEST(TestLogisticRegressionAlgorithm, TestMainLogisticRegression)
sq->record<kp::OpTensorSyncDevice>({ wIn, bIn });
sq->record<kp::OpAlgoBase>(
params, "test/shaders/glsl/test_logistic_regression.comp");
params, "test/shaders/glsl/test_logistic_regression.comp.spv");
sq->record<kp::OpTensorSyncLocal>({ wOutI, wOutJ, bOut, lOut });
@ -124,7 +124,7 @@ TEST(TestLogisticRegressionAlgorithm, TestMainLogisticRegressionManualCopy)
sq->begin();
sq->record<kp::OpAlgoBase>(
params, "test/shaders/glsl/test_logistic_regression.comp");
params, "test/shaders/glsl/test_logistic_regression.comp.spv");
sq->record<kp::OpTensorSyncLocal>({ wOutI, wOutJ, bOut, lOut });

3
test/shaders/glsl/test_op_custom_shader.comp
View file

@ -1,7 +1,10 @@
#version 450
layout (local_size_x = 1) in;
layout(set = 0, binding = 0) buffer a { float pa[]; };
layout(set = 0, binding = 1) buffer b { float pb[]; };
void main() {
uint index = gl_GlobalInvocationID.x;
pb[index] = pa[index];