1 // This file is part of Eigen, a lightweight C++ template library 2 // for linear algebra. 3 // 4 // Copyright (C) 2015 Vijay Vasudevan <vrv (at) google.com> 5 // 6 // This Source Code Form is subject to the terms of the Mozilla 7 // Public License v. 2.0. If a copy of the MPL was not distributed 8 // with this file, You can obtain one at http://mozilla.org/MPL/2.0/. 9 10 #define EIGEN_USE_THREADS 11 12 #include <stdlib.h> 13 #include "main.h" 14 #include <Eigen/CXX11/Tensor> 15 16 #if EIGEN_OS_WIN || EIGEN_OS_WIN64 17 #include <windows.h> 18 void sleep(int seconds) { 19 Sleep(seconds*1000); 20 } 21 #else 22 #include <unistd.h> 23 #endif 24 25 26 namespace { 27 28 void WaitAndAdd(Eigen::Notification* n, int* counter) { 29 n->Wait(); 30 *counter = *counter + 1; 31 } 32 33 } // namespace 34 35 static void test_notification_single() 36 { 37 ThreadPool thread_pool(1); 38 39 int counter = 0; 40 Eigen::Notification n; 41 std::function<void()> func = std::bind(&WaitAndAdd, &n, &counter); 42 thread_pool.Schedule(func); 43 sleep(1); 44 45 // The thread should be waiting for the notification. 46 VERIFY_IS_EQUAL(counter, 0); 47 48 // Unblock the thread 49 n.Notify(); 50 51 sleep(1); 52 53 // Verify the counter has been incremented 54 VERIFY_IS_EQUAL(counter, 1); 55 } 56 57 // Like test_notification_single() but enqueues multiple threads to 58 // validate that all threads get notified by Notify(). 59 static void test_notification_multiple() 60 { 61 ThreadPool thread_pool(1); 62 63 int counter = 0; 64 Eigen::Notification n; 65 std::function<void()> func = std::bind(&WaitAndAdd, &n, &counter); 66 thread_pool.Schedule(func); 67 thread_pool.Schedule(func); 68 thread_pool.Schedule(func); 69 thread_pool.Schedule(func); 70 sleep(1); 71 VERIFY_IS_EQUAL(counter, 0); 72 n.Notify(); 73 sleep(1); 74 VERIFY_IS_EQUAL(counter, 4); 75 } 76 77 void test_cxx11_tensor_notification() 78 { 79 CALL_SUBTEST(test_notification_single()); 80 CALL_SUBTEST(test_notification_multiple()); 81 } 82
