Update matrix_transpose.cl

Smart781 · Oct 6, 2024 · 149bc98 · 149bc98
1 parent 04f6ae4
commit 149bc98
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/cl/matrix_transpose.cl b/src/cl/matrix_transpose.cl
@@ -30,11 +30,11 @@ __kernel void matrix_transpose_local_bad_banks(
     int local_i = get_local_id(0);
     int local_j = get_local_id(1);
     if (i < M && j < K) {
-        tile[local_j][local_i] = A[i * K + j];
+        tile[local_j][local_i] = A[j * M + i];
     }
     barrier(CLK_LOCAL_MEM_FENCE);
     if (i < M && j < K) {
-        A_T[j * M + i] = tile[local_i][local_j];
+        A_T[i * K + j] = tile[local_i][local_j];
     }
 }