Adding warmup steps to coarsening.cu (#46)

gpu-mode · Jan 6, 2025 · 7314692 · 7314692
1 parent 32d3664
commit 7314692
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/lecture_008/coarsening.cu b/lecture_008/coarsening.cu
@@ -53,6 +53,13 @@ int main()
     cudaMemcpy(d_a, a, size, cudaMemcpyHostToDevice);
     cudaMemcpy(d_b, b, size, cudaMemcpyHostToDevice);
 
+    // warmup
+    VecAdd<<<(N + THREADS_PER_BLOCK - 1) / THREADS_PER_BLOCK, THREADS_PER_BLOCK>>>(d_a, d_b, d_c);
+    cudaDeviceSynchronize();
+
+    VecAddCoarsened<<<(N + 2*THREADS_PER_BLOCK - 1) / (2*THREADS_PER_BLOCK), THREADS_PER_BLOCK>>>(d_a, d_b, d_c);
+    cudaDeviceSynchronize();
+
     // Start timer for VecAdd kernel
     cudaEventRecord(start);
     VecAdd<<<(N + THREADS_PER_BLOCK - 1) / THREADS_PER_BLOCK, THREADS_PER_BLOCK>>>(d_a, d_b, d_c);