Update vectAdd kernel from idx to col

linjames0 · Sep 24, 2023 · d032783 · d032783
1 parent bfb0a12
commit d032783
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/vectAdd.cu b/vectAdd.cu
@@ -2,10 +2,10 @@
 #include <stdlib.h>
 
 __global__ void addVectors(float *d_a, float *d_b, float *d_c, int N) {
-        int i = threadIdx.x;
+        int col = blockIdx.x * blockDim.x + threadIdx.x;
 
-        if(i < N) {
-                d_c[i] = d_a[i] + d_b[i];
+        if(col < N) {
+                d_c[col] = d_a[col] + d_b[col];
         }
 }