Sat Mar 2 10:27:12 PM EST 2024

diff --git a/src/gpu.c b/src/gpu.c
index 7fb0f9647a297d935e522442b21e6a0e73f6ab52..b5e2ef013afcfbd5b07b9c0382ceee3a5d63c0b4 100644 (file)
--- a/src/gpu.c
+++ b/src/gpu.c
@@ -164,6 +164,12 @@ void GPU_mmul(float* ptrR, float* ptrA, float* ptrB, size_t rowsA, size_t colsB,
                fprintf(stderr, "GPU fatal error: clSetKernelArg() failed.\n");
                exit(1);
        }
+       err = clSetKernelArg(kernel, 5, sizeof(int), &shared);
+       if (err != CL_SUCCESS)
+       {
+               fprintf(stderr, "GPU fatal error: clSetKernelArg() failed.\n");
+               exit(1);
+       }
        //Run the program
        size_t work_size[] = {rowsA, colsB};
        err = clEnqueueNDRangeKernel(GPU_command_queue, kernel, 2, NULL, work_size, NULL, 0, NULL, NULL);