Backport: enforce memory order to avoid crash on Ampere (cms-sw#34386)

cms-patatrack · Aug 6, 2021 · f397769 · f397769
1 parent b86942b
commit f397769
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/RecoPixelVertexing/PixelTriplets/plugins/GPUCACell.h b/RecoPixelVertexing/PixelTriplets/plugins/GPUCACell.h
@@ -69,6 +69,7 @@ class GPUCACell {
       auto i = cellNeighbors.extend();  // maybe waisted....
       if (i > 0) {
         cellNeighbors[i].reset();
+        __threadfence();
 #ifdef __CUDACC__
         auto zero = (ptrAsInt)(&cellNeighbors[0]);
         atomicCAS((ptrAsInt*)(&theOuterNeighbors),
@@ -89,6 +90,7 @@ class GPUCACell {
       auto i = cellTracks.extend();  // maybe waisted....
       if (i > 0) {
         cellTracks[i].reset();
+        __threadfence();
 #ifdef __CUDACC__
         auto zero = (ptrAsInt)(&cellTracks[0]);
         atomicCAS((ptrAsInt*)(&theTracks), zero, (ptrAsInt)(&cellTracks[i]));  // if fails we cannot give "i" back...