Merge pull request #54 from exanauts/kk/dev

Fixing potential race conditions
exanauts · Aug 11, 2023 · 0075782 · 0075782
2 parents bfaa546 + 03c2332
commit 0075782
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/CUDA/ddot.jl b/src/CUDA/ddot.jl
@@ -9,5 +9,6 @@
     @inbounds for i=1:n
         v += dx[i]*dy[i]
     end
+    CUDA.sync_threads()
     return v
 end
diff --git a/src/CUDA/dspcg.jl b/src/CUDA/dspcg.jl
@@ -51,6 +51,7 @@
                 end
             end
         end
+        CUDA.sync_threads()
         nfree = CUDA.shfl_sync(0xffffffff, nfree, 1)
 
         # Exit if there are no free constraints.
-Original file line number
+Diff line change
@@ Expand Up / @@ -51,6 +51,7 @@ @@
                     end
                 end
             end
+            CUDA.sync_threads()
             nfree = CUDA.shfl_sync(0xffffffff, nfree, 1)
             # Exit if there are no free constraints.
@@ Expand Down @@