From e4f326e394e0454e536fc602967e74cfa398ef7a Mon Sep 17 00:00:00 2001
From: Alejandro Gallo <aamsgallo@gmail.com>
Date: Thu, 26 Jan 2023 01:50:34 +0100
Subject: [PATCH] Fix the reordering kernel in cuda

---
 src/atrip/Equations.cxx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/atrip/Equations.cxx b/src/atrip/Equations.cxx
index 6e88106..a324cfb 100644
--- a/src/atrip/Equations.cxx
+++ b/src/atrip/Equations.cxx
@@ -28,7 +28,7 @@ namespace atrip {
 #if defined(HAVE_CUDA)
 #define FOR_K()                                             \
   const size_t k = blockIdx.x * blockDim.x + threadIdx.x; \
-  size_t idx = 0;
+  size_t idx = k*size*size;
 #else
 #define FOR_K() for (size_t k=0, idx=0; k < size; k++)
 #endif