Port bugfix in Kokkos

2022-01-19 10:27:02 -07:00
parent 0a8cbcef2b
commit 6e8d9cb532
1 changed files with 1 additions and 1 deletions
--- a/lib/kokkos/core/src/Cuda/Kokkos_Cuda_Instance.cpp
+++ b/lib/kokkos/core/src/Cuda/Kokkos_Cuda_Instance.cpp
@ -694,7 +694,7 @@ std::pair<void *, int> CudaInternal::resize_team_scratch_space(
  int current_team_scratch = 0;
  int zero                 = 0;
  int one                  = 1;
-  while (m_team_scratch_pool[current_team_scratch].compare_exchange_weak(
+  while (!m_team_scratch_pool[current_team_scratch].compare_exchange_weak(
      zero, one, std::memory_order_release, std::memory_order_relaxed)) {
    current_team_scratch = (current_team_scratch + 1) % m_n_team_scratch;
  }