diff options
| author |  Eugene Zhulenev <ezhulenev@google.com> | 2019-10-16 17:14:37 -0700 |
|---|---|---|
| committer | Eugene Zhulenev <ezhulenev@google.com> | 2019-10-16 17:14:37 -0700 |
| commit | 0d2a14ce11c85abdfc68ca37fc66e3cace088b24 (patch) | |
| tree | b7838dd2a8dfe6a36f2c85bf9af58c2048df5e5b /unsupported/test/cxx11_tensor_block_eval.cpp | |
| parent | 02431cbe71eb036b1d6caa49c585db92a20b030f (diff) | |
Cleanup Tensor block destination and materialized block storage allocation
Diffstat (limited to 'unsupported/test/cxx11_tensor_block_eval.cpp')
| -rw-r--r-- | unsupported/test/cxx11_tensor_block_eval.cpp | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/unsupported/test/cxx11_tensor_block_eval.cpp b/unsupported/test/cxx11_tensor_block_eval.cpp index 01c431098..086dd8c11 100644 --- a/unsupported/test/cxx11_tensor_block_eval.cpp +++ b/unsupported/test/cxx11_tensor_block_eval.cpp @@ -154,9 +154,8 @@ static void VerifyBlockEvaluator(Expression expr, GenBlockParams gen_block) { Tensor<T, NumDims, Layout> dst(dst_dims); dst.setZero(); if (internal::random<bool>()) { - block_params.desc.template AddDestinationBuffer( - dst.data(), internal::strides<Layout>(dst.dimensions()), - dst.dimensions().TotalSize() * sizeof(T)); + block_params.desc.template AddDestinationBuffer<Layout>( + dst.data(), internal::strides<Layout>(dst.dimensions())); } const bool root_of_expr = internal::random<bool>(); |