     CubDebugExit(args.DeviceInit());
 
     // Get number of GPUs and current GPU
-    int num_gpus, initial_gpu;
+    int num_gpus;
+    int initial_gpu;
+    int timing_iterations           = 10000;
+    int timing_bytes                = 1024 * 1024;
+
     if (CubDebug(cudaGetDeviceCount(&num_gpus))) exit(1);
     if (CubDebug(cudaGetDevice(&initial_gpu))) exit(1);
+    args.GetCmdLineArgument("i", timing_iterations);
+    args.GetCmdLineArgument("bytes", timing_bytes);
 
     // Create default allocator (caches up to 6MB in device allocations per GPU)
     CachingDeviceAllocator allocator;

        AssertEquals(allocator.live_blocks.size(), 0);
    }
#endif  // CUB_CDP
#endif

    printf("Success\n");
    return 0;
}


    {
        printf("%s "
            "[--device=<device-id>]"
            "[--bytes=<timing bytes>]"
            "[--i=<timing iterations>]"
            "\n", argv[0]);
        exit(0);
    }
