diff options
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>(); |