Don't crash on empty tensor contraction.

(cherry picked from commit b0f877f8e01e90a5b0f3a79d46ea234899f8b499)
This commit is contained in:
Rasmus Munk Larsen 2023-04-05 17:06:14 +00:00 committed by Antonio Sanchez
parent 72b0759451
commit d4c24eca96

View File

@ -89,7 +89,6 @@ struct TensorContractionBlockMemAllocator {
eigen_assert(rhs_block);
BlockSizes sz = ComputeLhsRhsBlockSizes(bm, bk, bn);
char* block_mem = static_cast<char*>(d.allocate(sz.lhs_size + sz.rhs_size));
eigen_assert(block_mem);
*lhs_block = reinterpret_cast<LhsScalar*>(block_mem);
*rhs_block = reinterpret_cast<RhsScalar*>(block_mem + sz.lhs_size);
return block_mem;