mirror of
https://gitlab.com/libeigen/eigen.git
synced 2025-06-21 20:09:06 +08:00
Don't crash on empty tensor contraction.
This commit is contained in:
parent
15fbddaf9b
commit
b0f877f8e0
@ -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;
|
||||
|
Loading…
x
Reference in New Issue
Block a user