cory-master/BabelStream
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

You have to specify block size in CUDA, so make sure ARRAY_SIZE divides 1024

Browse Source
This commit is contained in:
Tom Deakin 2015-07-16 18:06:07 +01:00
parent 9da2cd6d09
commit 2c2dbf2c3f
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
cuda-stream.cu
View File

@ -131,6 +131,16 @@ int main(int argc, char *argv[])
<< "Version: " << VERSION_STRING << std::endl
<< "Implementation: CUDA" << std::endl << std::endl;
if (ARRAY_SIZE % 1024 != 0)
{
unsigned int OLD_ARRAY_SIZE = ARRAY_SIZE;
ARRAY_SIZE -= ARRAY_SIZE % 1024;
std::cout
<< "Warning: array size must divide 1024" << std::endl
<< "Resizing array from " << OLD_ARRAY_SIZE
<< " to " << ARRAY_SIZE << std::endl;
}
try
{
parseArguments(argc, argv);