Fix synchronization issue in stein (#775)

* Fix synchronization issue in stein * Updated changelog
ROCm · Aug 1, 2024 · ec88119 · ec88119
1 parent e169341
commit ec88119
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -49,6 +49,7 @@ Full documentation for rocSOLVER is available at the [rocSOLVER documentation](h
 
 ### Fixed
 - Fixed potential accuracy degradation in SYEVJ/HEEVJ for inputs with small eigenvalues.
+- Fixed synchronization issue in STEIN.
 
 ### Known Issues
 - A known issue in STEBZ can lead to errors in routines based on Bisection to compute eigenvalues for

diff --git a/library/src/auxiliary/rocauxiliary_stein.hpp b/library/src/auxiliary/rocauxiliary_stein.hpp
@@ -238,6 +238,7 @@ __device__ void run_stein(const int tid,
                 nrm2<MAX_THDS, S>(tid, blksize, work, 1, sval2);
                 __syncthreads();
                 scl = (work[sidx[0] - 1] >= 0 ? S(1) / sval2[0] : S(-1) / sval2[0]);
+                __syncthreads();
                 for(i = tid; i < blksize; i += MAX_THDS) // <- scal
                     work[i] = work[i] * scl;
                 __syncthreads();