GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-07-19 11:24:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

Don't crash on empty tensor contraction.

Browse Source 
(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
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
unsupported/Eigen/CXX11/src/Tensor/TensorContraction.h
View File

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