GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-08-15 13:15:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix tensor broadcast off-by-one error.

Browse Source 
Caught by JAX unit tests.  Triggered if broadcast is smaller than packet
size.


(cherry picked from commit ffb78e23a1b3bc232a07773144cfa5fa1759852d)
This commit is contained in:
Antonio Sanchez 2021-11-15 22:19:04 -08:00 committed by Rasmus Munk Larsen
parent c379a21191
commit 7e3bc4177e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
unsupported/Eigen/CXX11/src/Tensor/TensorBroadcasting.h
View File

@ -461,8 +461,8 @@ struct TensorEvaluator<const TensorBroadcastingOp<Broadcast, ArgType>, Device>
values[i] = m_impl.coeff(inputIndex); values[i] = m_impl.coeff(inputIndex);
++outputOffset; ++outputOffset;
} else { } else {
outputOffset = 0;
values[i] = m_impl.coeff(++inputIndex); values[i] = m_impl.coeff(++inputIndex);
outputOffset = 1; // Next offset.
} }
} }
return internal::pload<PacketReturnType>(values); return internal::pload<PacketReturnType>(values);