Updated default cuda example (#3685)

compiler-explorer · May 19, 2022 · f7f5f1c · f7f5f1c
1 parent 0954019
commit f7f5f1c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/cuda/default.cu b/examples/cuda/default.cu
@@ -1,6 +1,6 @@
 // Type your code here, or load an example.
-__global__ void square(int *array, int n) {
-    int tid = blockIdx.x;
+__global__ void square(int* array, int n) {
+    int tid = blockDim.x * blockIdx.x + threadIdx.x;
     if (tid < n)
         array[tid] = array[tid] * array[tid];
 }