mirror of
https://gitlab.com/libeigen/eigen.git
synced 2025-08-12 03:39:01 +08:00
Don't try to use direct offsets when computing a tensor product, since the required stride isn't available.
This commit is contained in:
parent
64ce78c2ec
commit
99cde88341
@ -344,7 +344,7 @@ class TensorContractionSubMapper {
|
|||||||
enum {
|
enum {
|
||||||
// We can use direct offsets iff the parent mapper supports then and we can compute the strides.
|
// We can use direct offsets iff the parent mapper supports then and we can compute the strides.
|
||||||
// TODO: we should also enable direct offsets for the Rhs case.
|
// TODO: we should also enable direct offsets for the Rhs case.
|
||||||
UseDirectOffsets = (side == Lhs) && inner_dim_contiguous && ParentMapper::DirectOffsets
|
UseDirectOffsets = ParentMapper::DirectOffsets && (side == Lhs) && inner_dim_contiguous && (array_size<contract_t>::value > 0)
|
||||||
};
|
};
|
||||||
|
|
||||||
EIGEN_DEVICE_FUNC TensorContractionSubMapper(const ParentMapper& base_mapper, Index vert_offset, Index horiz_offset)
|
EIGEN_DEVICE_FUNC TensorContractionSubMapper(const ParentMapper& base_mapper, Index vert_offset, Index horiz_offset)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user