From 0d2a14ce11c85abdfc68ca37fc66e3cace088b24 Mon Sep 17 00:00:00 2001
From: Eugene Zhulenev <ezhulenev@google.com>
Date: Wed, 16 Oct 2019 17:14:37 -0700
Subject: Cleanup Tensor block destination and materialized block storage
 allocation

---
 unsupported/test/cxx11_tensor_block_eval.cpp | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

(limited to 'unsupported/test/cxx11_tensor_block_eval.cpp')

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>();
-- 
cgit v1.2.3