Sat Mar 2 10:20:16 PM EST 2024

diff --git a/src/gpu.c b/src/gpu.c
index f9fbfe5e49284c797ed95cfef9113d4b419bd3fd..7fb0f9647a297d935e522442b21e6a0e73f6ab52 100644 (file)
--- a/src/gpu.c
+++ b/src/gpu.c
@@ -108,13 +108,14 @@ void GPU_mmul(float* ptrR, float* ptrA, float* ptrB, size_t rowsA, size_t colsB,
        memcpy(tmp, src_gpu_mmul_cl, src_gpu_mmul_cl_len);
        const char* ptr = (const char*)src_gpu_mmul_cl;
        cl_program program = clCreateProgramWithSource(GPU_context, 1, (const char**)&tmp, NULL, &err);
-       free(tmp);
        if (err != CL_SUCCESS)
        {
+               free(tmp);
                fprintf(stderr, "GPU fatal error: clCreateProgramWithSource() failed.\n");
                exit(1);
        }
        err = clBuildProgram(program, 1, &GPU_device_id, NULL, NULL, NULL);
+       free(tmp);
        if (err != CL_SUCCESS)
        {
                fprintf(stderr, "GPU fatal error: clBuildProgram() failed.\n");