devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add AVX512 s/dgemm optimizations for compute kernel (2nd try)

Browse Source
This commit is contained in:
aaraujom
2022-05-28 02:00:21 +00:00
committed by Antonio Sánchez
parent 510f6b9f15
commit d49ede4dc4
9 changed files with 1377 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
Eigen/src/Core/util/BlasUtil.h
View File

@@ -229,6 +229,7 @@ public:
}
EIGEN_DEVICE_FUNC const Index stride() const { return m_stride; }
EIGEN_DEVICE_FUNC const Index incr() const { return 1; }
EIGEN_DEVICE_FUNC const Scalar* data() const { return m_data; }
EIGEN_DEVICE_FUNC Index firstAligned(Index size) const {
@@ -402,6 +403,10 @@ public:
storePacketBlock_helper<SubPacket, Scalar, n, n-1> spb;
spb.store(this, i,j,block);
}
EIGEN_DEVICE_FUNC const Index stride() const { return m_stride; }
EIGEN_DEVICE_FUNC const Index incr() const { return m_incr.value(); }
EIGEN_DEVICE_FUNC Scalar* data() const { return m_data; }
protected:
Scalar* EIGEN_RESTRICT m_data;
const Index m_stride;