Inherit alignment trait from argument in TensorBroadcasting to avoid segfault when the argument is unaligned.

This commit is contained in:
Rasmus Munk Larsen 2020-07-28 19:19:37 +00:00
parent 99da2e1a8d
commit b92206676c

View File

@ -112,7 +112,7 @@ struct TensorEvaluator<const TensorBroadcastingOp<Broadcast, ArgType>, Device>
typedef typename Storage::Type EvaluatorPointerType;
enum {
IsAligned = true,
IsAligned = TensorEvaluator<ArgType, Device>::IsAligned,
PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess,
BlockAccess = TensorEvaluator<ArgType, Device>::BlockAccess,
PreferBlockAccess = true,