Don't crash on empty tensor contraction.

This commit is contained in:
Rasmus Munk Larsen 2023-04-05 17:06:14 +00:00
parent 15fbddaf9b
commit b0f877f8e0

View File

@ -91,7 +91,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;