gallo/atrip
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix the reordering kernel in cuda

Browse Source
This commit is contained in:
Alejandro Gallo 2023-01-26 01:50:34 +01:00
parent 93cba3c3ab
commit e4f326e394
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/atrip/Equations.cxx
View File

@ -28,7 +28,7 @@ namespace atrip {
#if defined(HAVE_CUDA) #if defined(HAVE_CUDA)
#define FOR_K() \ #define FOR_K() \
const size_t k = blockIdx.x * blockDim.x + threadIdx.x; \ const size_t k = blockIdx.x * blockDim.x + threadIdx.x; \
size_t idx = 0; size_t idx = k*size*size;
#else #else
#define FOR_K() for (size_t k=0, idx=0; k < size; k++) #define FOR_K() for (size_t k=0, idx=0; k < size; k++)
#endif #endif