Fix a bug in a packed block type in TensorContractionThreadPool

1 files changed, 2 insertions, 2 deletions

diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorContractionThreadPool.h b/unsupported/Eigen/CXX11/src/Tensor/TensorContractionThreadPool.h
index 0f091b42f..873db5efd 100644
--- a/unsupported/Eigen/CXX11/src/Tensor/TensorContractionThreadPool.h
+++ b/unsupported/Eigen/CXX11/src/Tensor/TensorContractionThreadPool.h
@@ -750,7 +750,7 @@ struct TensorEvaluator<const TensorContractionOp<Indices, LeftArgType, RightArgT
       template <typename EvalCtx>
       struct ThreadLocalBlocksAllocator</*pack_rhs=*/false, EvalCtx> {
         static void allocate(EvalCtx& ctx, Blocks& blocks) {
-          std::vector<RhsBlock> lhs_blocks;
+          std::vector<LhsBlock> lhs_blocks;
           BlockMemHandle mem_handle = ctx.kernel_.allocateSlices(
               ctx.device_,
               /*num_lhs=*/ctx.gm_,
@@ -758,7 +758,7 @@ struct TensorEvaluator<const TensorContractionOp<Indices, LeftArgType, RightArgT
               /*num_slices=*/1,
               /*lhs_blocks=*/&lhs_blocks, /*rhs_blocks=*/nullptr);
 
-          blocks = ThreadLocalBlocks<RhsBlock>(std::move(mem_handle),
+          blocks = ThreadLocalBlocks<LhsBlock>(std::move(mem_handle),
                                                std::move(lhs_blocks));
         }