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

Enable extract et. al. for HIP GPU.

Browse Source
This commit is contained in:
Rohit Santhanam 2021-07-09 14:58:07 +00:00
parent 0c361c4899
commit beea14a18f
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -518,7 +518,7 @@ struct blas_traits<const T>
template<typename T, bool HasUsableDirectAccess=blas_traits<T>::HasUsableDirectAccess> template<typename T, bool HasUsableDirectAccess=blas_traits<T>::HasUsableDirectAccess>
struct extract_data_selector { struct extract_data_selector {
static const typename T::Scalar* run(const T& m) EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE static const typename T::Scalar* run(const T& m)
{ {
return blas_traits<T>::extract(m).data(); return blas_traits<T>::extract(m).data();
} }
@ -529,7 +529,8 @@ struct extract_data_selector<T,false> {
static typename T::Scalar* run(const T&) { return 0; } static typename T::Scalar* run(const T&) { return 0; }
}; };
template<typename T> const typename T::Scalar* extract_data(const T& m) template<typename T>
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE const typename T::Scalar* extract_data(const T& m)
{ {
return extract_data_selector<T>::run(m); return extract_data_selector<T>::run(m);
} }