mirror of
https://gitlab.com/libeigen/eigen.git
synced 2025-06-22 07:22:28 +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);
|
eigen_assert(rhs_block);
|
||||||
BlockSizes sz = ComputeLhsRhsBlockSizes(bm, bk, bn);
|
BlockSizes sz = ComputeLhsRhsBlockSizes(bm, bk, bn);
|
||||||
char* block_mem = static_cast<char*>(d.allocate(sz.lhs_size + sz.rhs_size));
|
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);
|
*lhs_block = reinterpret_cast<LhsScalar*>(block_mem);
|
||||||
*rhs_block = reinterpret_cast<RhsScalar*>(block_mem + sz.lhs_size);
|
*rhs_block = reinterpret_cast<RhsScalar*>(block_mem + sz.lhs_size);
|
||||||
return block_mem;
|
return block_mem;
|
||||||
|
Loading…
x
Reference in New Issue
Block a user