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

Zero Tijk correctly in CPU code

Browse Source
This commit is contained in:
Ania Brown 2023-01-23 16:58:08 +01:00 committed by Alejandro Gallo
parent 3fe15e5e5c
commit 58c0bf078e
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/atrip/Equations.cxx
View File

@ -558,12 +558,14 @@ void getEnergySame
acc::zeroing<<<bs, ths>>>((DataFieldType<F>*)Tijk, acc::zeroing<<<bs, ths>>>((DataFieldType<F>*)Tijk,
NoNoNo); NoNoNo);
) )
#else #endif
#if !defined(HAVE_CUDA)
WITH_CHRONO("double:reorder", WITH_CHRONO("double:reorder",
for (size_t k = 0; k < NoNoNo; k++) { for (size_t k = 0; k < NoNoNo; k++) {
Tijk[k] = DataFieldType<F>{0.0}; Tijk[k] = DataFieldType<F>{0.0};
}) })
#endif /* defined(HAVE_CUDA) && !defined(ATRIP_ONLY_DGEMM) */ #endif /* !defined(HAVE_CUDA) */
#if defined(ATRIP_ONLY_DGEMM) #if defined(ATRIP_ONLY_DGEMM)