diff options
author | Eugene Zhulenev <eugene.zhulenev@gmail.com> | 2019-12-18 20:07:00 +0000 |
---|---|---|
committer | Rasmus Munk Larsen <rmlarsen@google.com> | 2019-12-18 20:07:00 +0000 |
commit | ae07801dd8d295657f28b006e1e4999edf835052 (patch) | |
tree | 08a91a4368c15d365127344f920bd10f8e437db2 /unsupported/test/cxx11_tensor_block_io.cpp | |
parent | 72166d0e6eaf12a99f449e26f402f926bef2bb50 (diff) |
Tensor block evaluation cost model
Diffstat (limited to 'unsupported/test/cxx11_tensor_block_io.cpp')
-rw-r--r-- | unsupported/test/cxx11_tensor_block_io.cpp | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/unsupported/test/cxx11_tensor_block_io.cpp b/unsupported/test/cxx11_tensor_block_io.cpp index 25584433e..b8600eaea 100644 --- a/unsupported/test/cxx11_tensor_block_io.cpp +++ b/unsupported/test/cxx11_tensor_block_io.cpp @@ -75,8 +75,8 @@ static void test_block_io_copy_data_from_source_to_target() { // Construct a tensor block mapper. using TensorBlockMapper = internal::TensorBlockMapper<NumDims, Layout, Index>; - TensorBlockMapper block_mapper(dims, {RandomBlockShape(), - RandomTargetBlockSize(dims)}); + TensorBlockMapper block_mapper( + dims, {RandomBlockShape(), RandomTargetBlockSize(dims), {0, 0, 0}}); // We will copy data from input to output through this buffer. Tensor<T, NumDims, Layout> block(block_mapper.blockDimensions()); @@ -146,8 +146,10 @@ static void test_block_io_copy_using_reordered_dimensions() { // NOTE: Tensor block mapper works with shuffled dimensions. using TensorBlockMapper = internal::TensorBlockMapper<NumDims, Layout, Index>; - TensorBlockMapper block_mapper(output_tensor_dims, {RandomBlockShape(), - RandomTargetBlockSize(output_tensor_dims)}); + TensorBlockMapper block_mapper(output_tensor_dims, + {RandomBlockShape(), + RandomTargetBlockSize(output_tensor_dims), + {0, 0, 0}}); // We will copy data from input to output through this buffer. Tensor<T, NumDims, Layout> block(block_mapper.blockDimensions()); |