Fix SimpleBarrier
[folly.git] / folly / test / SmallLocksBenchmark.cpp
index 9b86adf..ca69b1a 100644 (file)
@@ -51,7 +51,7 @@ struct SimpleBarrier {
     if (++num_ == count_) {
       cv_.notify_all();
     } else {
     if (++num_ == count_) {
       cv_.notify_all();
     } else {
-      cv_.wait(lockHeld);
+      cv_.wait(lockHeld, [&]() { return num_ >= count_; });
     }
   }
 
     }
   }