diff options
| author |  Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-01-14 16:39:39 -0800 |
|---|---|---|
| committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-01-14 16:39:39 -0800 |
| commit | a282eb13632e701451eac4f89858ddcd122f1254 (patch) | |
| tree | a33d699d3b838ef57b1c794259e3510e1710a97f /Eigen | |
| parent | 7832485575a53ca81fa29d21fc5bf57dc8973e0e (diff) | |
pscatter/pgather use Index instead of int to specify the stride
Diffstat (limited to 'Eigen')
| -rw-r--r-- | Eigen/src/Core/arch/AVX512/PacketMath.h | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/Eigen/src/Core/arch/AVX512/PacketMath.h b/Eigen/src/Core/arch/AVX512/PacketMath.h index 279b0b3ff..971aaca7f 100644 --- a/Eigen/src/Core/arch/AVX512/PacketMath.h +++ b/Eigen/src/Core/arch/AVX512/PacketMath.h @@ -550,7 +550,7 @@ EIGEN_STRONG_INLINE void pstoreu<int>(int* to, const Packet16i& from) { template <> EIGEN_DEVICE_FUNC inline Packet16f pgather<float, Packet16f>(const float* from, - int stride) { + Index stride) { Packet16i stride_vector = _mm512_set1_epi32(stride); Packet16i stride_multiplier = _mm512_set_epi32(15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0); @@ -560,7 +560,7 @@ EIGEN_DEVICE_FUNC inline Packet16f pgather<float, Packet16f>(const float* from, } template <> EIGEN_DEVICE_FUNC inline Packet8d pgather<double, Packet8d>(const double* from, - int stride) { + Index stride) { Packet8i stride_vector = _mm256_set1_epi32(stride); Packet8i stride_multiplier = _mm256_set_epi32(7, 6, 5, 4, 3, 2, 1, 0); Packet8i indices = _mm256_mullo_epi32(stride_vector, stride_multiplier); @@ -571,7 +571,7 @@ EIGEN_DEVICE_FUNC inline Packet8d pgather<double, Packet8d>(const double* from, template <> EIGEN_DEVICE_FUNC inline void pscatter<float, Packet16f>(float* to, const Packet16f& from, - int stride) { + Index stride) { Packet16i stride_vector = _mm512_set1_epi32(stride); Packet16i stride_multiplier = _mm512_set_epi32(15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0); @@ -581,7 +581,7 @@ EIGEN_DEVICE_FUNC inline void pscatter<float, Packet16f>(float* to, template <> EIGEN_DEVICE_FUNC inline void pscatter<double, Packet8d>(double* to, const Packet8d& from, - int stride) { + Index stride) { Packet8i stride_vector = _mm256_set1_epi32(stride); Packet8i stride_multiplier = _mm256_set_epi32(7, 6, 5, 4, 3, 2, 1, 0); Packet8i indices = _mm256_mullo_epi32(stride_vector, stride_multiplier); |