From ac89651ff12017f8eecfa7470fb09d432a5dc315 Mon Sep 17 00:00:00 2001 From: miha-q <> Date: Sat, 2 Mar 2024 22:37:45 -0500 Subject: [PATCH] Sat Mar 2 10:37:45 PM EST 2024 --- src/gpu_mmul.cl | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/src/gpu_mmul.cl b/src/gpu_mmul.cl index 1f8e4f8..31cada3 100644 --- a/src/gpu_mmul.cl +++ b/src/gpu_mmul.cl @@ -8,14 +8,17 @@ __kernel void gpu_mmul const int shared ) { - const int colsA = shared; - const int rowsB = shared; + int colsA = shared; + int rowsB = shared; + int rowsR = rowsA; + int colsR = colsB; int row = get_global_id(0); int col = get_global_id(1); float sum = 0; - for (int i = 0; i < rowsA; i++) + + for (int i = 0; i < colsR; i++) { sum += ptrA[row * colsA + i] * ptrB[i * colsB + col]; } - ptrR[row * colsB + col] = sum; + ptrR[row * colsR + col] = sum; } \ No newline at end of file -- 2.39.5