stop being dumb dumb

shaders/A2Task1Interleaved.comp

@@ -17,7 +17,11 @@ layout(push_constant) uniform PushStruct {
 layout(binding = 0) buffer inoutBufer {uint v[];};
 
 void main() {
-    uint idx = gl_GlobalInvocationID.x;
+    uint idx = gl_GlobalInvocationID.x * 2 * p.stride;
-    
+
+    if (idx + p.stride >= p.size) {
+        return;
+    }
+
     v[idx] += v[idx + p.stride];
 }

shaders/A2Task1Sequential.comp

@@ -18,9 +18,10 @@ layout(binding = 0) buffer inoutBufer { uint v[]; };
 
 void main() {    
     uint gIDx = gl_GlobalInvocationID.x;
-    if (gIDx >= p.size) {
+    
+    if (gIDx + p.offset >= p.size) {
         return;
     }
 
-    v[gIDx] = v[gIDx] + v[gIDx + p.offset];
+    v[gIDx] += v[gIDx + p.offset];
 }

src/A2Task1Solution/Interleaved.cpp

@@ -56,12 +56,12 @@ void A2Task1SolutionInterleaved::compute()
     PushConstant pc;
     pc.size = inputSize;
 
-    for (uint stride = inputSize / 2; stride > 0; stride >>= 1)
+    for (uint stride = 1; stride < inputSize; stride <<= 1)
     {
         pc.stride = stride;
         cb.pushConstants(pipelineLayout, vk::ShaderStageFlagBits::eCompute, 0, sizeof(PushConstant), &pc);
 
-        cb.dispatch((stride + workGroupSize - 1) / workGroupSize, 1, 1);
+        cb.dispatch((inputSize / (stride * 2) + workGroupSize - 1) / workGroupSize, 1, 1);
 
         vk::MemoryBarrier memoryBarrier(vk::AccessFlagBits::eShaderWrite, vk::AccessFlagBits::eShaderRead);
         cb.pipelineBarrier(vk::PipelineStageFlagBits::eComputeShader, vk::PipelineStageFlagBits::eComputeShader,

src/A2Task1Solution/Sequential.cpp

@@ -62,7 +62,7 @@ void A2Task1SolutionSequential::compute()
     {
         cb.pushConstants(pipelineLayout, vk::ShaderStageFlagBits::eCompute, 0, sizeof(PushConstant), &pc);
 
-        uint groupCount = (pc.size + pc.offset + workGroupSize - 1) / workGroupSize;
+        uint groupCount = (pc.offset + workGroupSize - 1) / workGroupSize;
         cb.dispatch(groupCount, 1, 1);
 
         vk::MemoryBarrier memoryBarrier(vk::AccessFlagBits::eShaderWrite, vk::AccessFlagBits::eShaderRead);