 
     dim3 dim_block(THPB, 1, 1);
     dim3 dim_grid(numBlocks, 1, 1);
-    printf("\nDimgrid : %d \n", numBlocks);
+    // printf("\nDimgrid : %d \n", numBlocks);
 
     // computes the prefix sum per block
     pref_sum_parallel <<< dim_grid, dim_block >>> (dA, dB, n);

    CUDA_CHECK();

    CUDA_CALL( cudaMemcpy(ans, dB, n * sizeof(int), cudaMemcpyDeviceToHost) );
    CUDA_CALL( cudaMemcpy(pref_b, dCT, numBlocks * sizeof(int), cudaMemcpyDeviceToHost) );
    // printf("\n -- %d -- \n", cudaMemcpy(ans, dB, n * sizeof(int), cudaMemcpyDeviceToHost));

#if 1
    for (int i = 0; i < n; ++i) {
      printf("%d ", A[i]);
    }

    if (!ok) {
      printf("Problem with prefix sum on test case : %d, element %d\n", tc, idx);
      printf("%d != %d\n", B[idx], ans[idx]);
      exit(1);
    }

