devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add block evaluation to TensorReshaping/TensorCasting/TensorPadding/TensorSelect

Browse Source
This commit is contained in:
Eugene Zhulenev
2019-10-02 12:44:06 -07:00
parent 6e40454a6e
commit 60ae24ee1a
9 changed files with 862 additions and 167 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
unsupported/test/cxx11_tensor_executor.cpp
View File

@@ -582,11 +582,10 @@ static void test_async_execute_unary_expr(Device d)
Eigen::Barrier done(1);
auto on_done = [&done]() { done.Notify(); };
static const bool TilingOn = Tiling == TiledEvaluation::Off ? false : true;
using Assign = TensorAssignOp<decltype(dst), const decltype(expr)>;
using DoneCallback = decltype(on_done);
using Executor = internal::TensorAsyncExecutor<const Assign, Device, DoneCallback,
Vectorizable, TilingOn>;
Vectorizable, Tiling>;
Executor::runAsync(Assign(dst, expr), d, on_done);
done.Wait();
@@ -619,11 +618,10 @@ static void test_async_execute_binary_expr(Device d)
Eigen::Barrier done(1);
auto on_done = [&done]() { done.Notify(); };
static const bool TilingOn = Tiling == TiledEvaluation::Off ? false : true;
using Assign = TensorAssignOp<decltype(dst), const decltype(expr)>;
using DoneCallback = decltype(on_done);
using Executor = internal::TensorAsyncExecutor<const Assign, Device, DoneCallback,
Vectorizable, TilingOn>;
Vectorizable, Tiling>;
Executor::runAsync(Assign(dst, expr), d, on_done);
done.Wait();
@@ -737,10 +735,10 @@ EIGEN_DECLARE_TEST(cxx11_tensor_executor) {
CALL_SUBTEST_COMBINATIONS_V1(8, test_execute_reduction, float, 4);
CALL_SUBTEST_COMBINATIONS_V1(8, test_execute_reduction, float, 5);
CALL_SUBTEST_COMBINATIONS_V1(9, test_execute_reshape, float, 2);
CALL_SUBTEST_COMBINATIONS_V1(9, test_execute_reshape, float, 3);
CALL_SUBTEST_COMBINATIONS_V1(9, test_execute_reshape, float, 4);
CALL_SUBTEST_COMBINATIONS_V1(9, test_execute_reshape, float, 5);
CALL_SUBTEST_COMBINATIONS_V2(9, test_execute_reshape, float, 2);
CALL_SUBTEST_COMBINATIONS_V2(9, test_execute_reshape, float, 3);
CALL_SUBTEST_COMBINATIONS_V2(9, test_execute_reshape, float, 4);
CALL_SUBTEST_COMBINATIONS_V2(9, test_execute_reshape, float, 5);
CALL_SUBTEST_COMBINATIONS_V1(10, test_execute_slice_rvalue, float, 2);
CALL_SUBTEST_COMBINATIONS_V1(10, test_execute_slice_rvalue, float, 3);
@@ -779,4 +777,3 @@ EIGEN_DECLARE_TEST(cxx11_tensor_executor) {
// Force CMake to split this test.
// EIGEN_SUFFIXES;1;2;3;4;5;6;7;8;9;10;11;12;13;14;15;16
}