Mercurial > hg > Members > Moririn
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]); }