GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-06-04 18:54:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix data race in css11_tensor_notification test.

Browse Source
This commit is contained in:
Rasmus Munk Larsen 2019-11-08 17:44:50 -08:00
parent cc3d0e6a40
commit ebf04fb3e8
1 changed files with 6 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
unsupported/test/cxx11_tensor_notification.cpp
View File

@ -9,27 +9,19 @@
#define EIGEN_USE_THREADS
#include <atomic>
#include <stdlib.h>
#include "main.h"
#include <Eigen/CXX11/Tensor>
namespace {
void WaitAndAdd(Eigen::Notification* n, int* counter) {
n->Wait();
*counter = *counter + 1;
}
} // namespace
static void test_notification_single()
{
ThreadPool thread_pool(1);
int counter = 0;
std::atomic<int> counter(0);
Eigen::Notification n;
std::function<void()> func = std::bind(&WaitAndAdd, &n, &counter);
auto func = [&n, &counter](){ n.Wait(); ++counter;};
thread_pool.Schedule(func);
EIGEN_SLEEP(1000);
@ -51,9 +43,9 @@ static void test_notification_multiple()
{
ThreadPool thread_pool(1);
int counter = 0;
std::atomic<int> counter(0);
Eigen::Notification n;
std::function<void()> func = std::bind(&WaitAndAdd, &n, &counter);
auto func = [&n, &counter](){ n.Wait(); ++counter;};
thread_pool.Schedule(func);
thread_pool.Schedule(func);
thread_pool.Schedule(func);