mirror/eigen
2
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2024-12-21 07:19:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

Made sure to use a tensor of rank 0 to store the result of a full reduction in the tensor thread pool test

Browse Source
This commit is contained in:
Benoit Steiner 2016-01-30 10:46:36 -08:00
parent d0db95f730
commit 2053478c56
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
unsupported/test/cxx11_tensor_thread_pool.cpp
View File

@ -239,15 +239,15 @@ void test_multithreaded_reductions() {
Tensor<float, 2, DataLayout> t1(num_rows, num_cols);
t1.setRandom();
Tensor<float, 1, DataLayout> full_redux(1);
Tensor<float, 0, DataLayout> full_redux;
full_redux = t1.sum();
Tensor<float, 1, DataLayout> full_redux_tp(1);
Tensor<float, 0, DataLayout> full_redux_tp;
full_redux_tp.device(thread_pool_device) = t1.sum();
// Check that the single threaded and the multi threaded reductions return
// the same result.
VERIFY_IS_APPROX(full_redux(0), full_redux_tp(0));
VERIFY_IS_APPROX(full_redux(), full_redux_tp());
}