Skip to content
Snippets Groups Projects
Commit 695575ce authored by Tomáš Oberhuber's avatar Tomáš Oberhuber
Browse files

Fixing CUDA memset value.

parent dcf62dfd
No related branches found
No related tags found
No related merge requests found
...@@ -87,21 +87,10 @@ bool setMemoryCuda( Element* data, ...@@ -87,21 +87,10 @@ bool setMemoryCuda( Element* data,
dim3 blockSize, gridSize; dim3 blockSize, gridSize;
blockSize. x = 512; blockSize. x = 512;
int blocksNumber = ceil( ( double ) size / ( double ) blockSize. x ); int blocksNumber = ceil( ( double ) size / ( double ) blockSize. x );
int gridsNumber = ceil( ( double ) blocksNumber / ( double ) maxCudaGridSize ); int elementsPerThread = ceil( ( double ) blocksNumber / ( double ) maxCudaGridSize );
setVectorValueCudaKernel<<< gridSize, blockSize >>>( data, size, value, elementsPerThread );
cerr << "Grids number is " << gridsNumber << endl;
for( int gridIdx = 0; gridIdx < gridsNumber; gridIdx ++ )
{
cout << "GridIdx = " << gridIdx << endl;
if( gridIdx < gridsNumber - 1 )
gridSize. x = maxCudaGridSize;
else
gridSize. x = blocksNumber - gridIdx * maxCudaGridSize;
cout << "GridSize. x = " << gridSize. x << endl;
setVectorValueCudaKernel<<< gridSize, blockSize >>>
( &( data[ gridIdx * maxCudaGridSize * blockSize. x ] ),
value );
}
return checkCudaDevice; return checkCudaDevice;
#else #else
cerr << "I am sorry but CUDA support is missing on this system " << __FILE__ << " line " << __LINE__ << "." << endl; cerr << "I am sorry but CUDA support is missing on this system " << __FILE__ << " line " << __LINE__ << "." << endl;
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment