diff options
author | Eugene Zhulenev <ezhulenev@google.com> | 2019-10-04 10:15:33 -0700 |
---|---|---|
committer | Eugene Zhulenev <ezhulenev@google.com> | 2019-10-04 10:15:33 -0700 |
commit | 98bdd7252e14441dc2d392c5146496c35d5e6062 (patch) | |
tree | 96e61318d44c047621b0a40fe496548e93397bd6 /unsupported/test | |
parent | 60ae24ee1a6c16114de456d77fcfba6f5a1160ca (diff) |
Fix compilation warnings and errors with clang in TensorBlockV2 code and tests
Diffstat (limited to 'unsupported/test')
-rw-r--r-- | unsupported/test/cxx11_tensor_block_eval.cpp | 8 | ||||
-rw-r--r-- | unsupported/test/cxx11_tensor_executor.cpp | 6 |
2 files changed, 8 insertions, 6 deletions
diff --git a/unsupported/test/cxx11_tensor_block_eval.cpp b/unsupported/test/cxx11_tensor_block_eval.cpp index ff98e170d..75252362c 100644 --- a/unsupported/test/cxx11_tensor_block_eval.cpp +++ b/unsupported/test/cxx11_tensor_block_eval.cpp @@ -104,13 +104,13 @@ static TensorBlockParams<NumDims> FixedSizeBlock(DSizes<Index, NumDims> dims) { return {offsets, dims, TensorBlockDescriptor<NumDims, Index>(0, dims)}; } -inline Eigen::IndexList<int, Eigen::type2index<1>> NByOne(int n) { - Eigen::IndexList<int, Eigen::type2index<1>> ret; +inline Eigen::IndexList<Index, Eigen::type2index<1>> NByOne(Index n) { + Eigen::IndexList<Index, Eigen::type2index<1>> ret; ret.set(0, n); return ret; } -inline Eigen::IndexList<Eigen::type2index<1>, int> OneByM(int m) { - Eigen::IndexList<Eigen::type2index<1>, int> ret; +inline Eigen::IndexList<Eigen::type2index<1>, Index> OneByM(Index m) { + Eigen::IndexList<Eigen::type2index<1>, Index> ret; ret.set(1, m); return ret; } diff --git a/unsupported/test/cxx11_tensor_executor.cpp b/unsupported/test/cxx11_tensor_executor.cpp index 9094b6507..efae81961 100644 --- a/unsupported/test/cxx11_tensor_executor.cpp +++ b/unsupported/test/cxx11_tensor_executor.cpp @@ -582,10 +582,11 @@ 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, Tiling>; + Vectorizable, TilingOn>; Executor::runAsync(Assign(dst, expr), d, on_done); done.Wait(); @@ -618,10 +619,11 @@ 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, Tiling>; + Vectorizable, TilingOn>; Executor::runAsync(Assign(dst, expr), d, on_done); done.Wait(); |