     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
+#if 0
     for (int i = 0; i < n; ++i) {
       printf("%d ", A[i]);
     }


const int MV = 15;
int main() {
  int lengths[] = {16, 1024, 12288, 1048576, 10002432}; // 2048 * X <= 10002432
  srand(time(0));
  for (int tc = 0; tc < 1; ++tc) {
    int n = lengths[tc];
    int *A = new int[n], *B = new int[n];
    for (int i = 0; i < n; ++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);
    }

