diff src/test/twice.cc @ 301:609bf62768b9

add -DUSE_CUDA=1 flag to cmake
author Shinji KONO <kono@ie.u-ryukyu.ac.jp>
date Sun, 12 Feb 2017 12:35:11 +0900
parents 8bbc0012e1a4
children 1839586f5b41
line wrap: on
line diff
--- a/src/test/twice.cc	Sun Feb 12 09:12:21 2017 +0900
+++ b/src/test/twice.cc	Sun Feb 12 12:35:11 2017 +0900
@@ -112,13 +112,13 @@
     for (int i=0;i<num_exec;i++,cur++) {
         if (num_stream <= cur)
             cur=0;
-        B[i] = (float)(i+1);
+        //B[i] = (float)(i+1);
         //cuMemcpyHtoDAsync(devB[i], &B[i], sizeof(float), stream[cur]);
         void* args[] = {&devA, &devB[i], &devOut[i]};
         checkCudaErrors(cuLaunchKernel(function,
                        LENGTH, 1, 1,
                        THREAD, 1, 1,
-                                 0, stream[cur], args, NULL));
+                                 0, num_stream ? stream[cur] : NULL , args, NULL));
         //cuMemcpyDtoHAsync(result[i], devOut[i], LENGTH*THREAD*sizeof(float), stream[cur]);
     }