Merge pull request #35238 from ROCmSoftwarePlatform/r2.1-eigen-patch
[ROCm][r2.1] eigen patch needed for HIP header changes
This commit is contained in:
Goldie Gadde
GitHub
commit
9837eceb39
157
third_party/eigen3/gpu_packet_math.patch
vendored
157
third_party/eigen3/gpu_packet_math.patch
vendored
@ -22,4 +22,161 @@
|
|||||||
return res;
|
return res;
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
--- a/unsupported/Eigen/SpecialFunctions
|
||||||
|
+++ b/unsupported/Eigen/SpecialFunctions
|
||||||
|
@@ -48,6 +48,9 @@
|
||||||
|
}
|
||||||
|
|
||||||
|
#include "src/SpecialFunctions/SpecialFunctionsImpl.h"
|
||||||
|
+#if defined(EIGEN_HIPCC)
|
||||||
|
+#include "src/SpecialFunctions/HipVectorCompatibility.h"
|
||||||
|
+#endif
|
||||||
|
#include "src/SpecialFunctions/SpecialFunctionsPacketMath.h"
|
||||||
|
#include "src/SpecialFunctions/SpecialFunctionsHalf.h"
|
||||||
|
#include "src/SpecialFunctions/SpecialFunctionsFunctors.h"
|
||||||
|
--- /dev/null
|
||||||
|
+++ b/unsupported/Eigen/src/SpecialFunctions/HipVectorCompatibility.h
|
||||||
|
@@ -0,0 +1,143 @@
|
||||||
|
+#ifndef HIP_VECTOR_COMPATIBILITY_H
|
||||||
|
+#define HIP_VECTOR_COMPATIBILITY_H
|
||||||
|
+
|
||||||
|
+namespace hip_impl {
|
||||||
|
+ template <typename, typename, unsigned int> struct Scalar_accessor;
|
||||||
|
+} // end namespace hip_impl
|
||||||
|
+
|
||||||
|
+namespace Eigen {
|
||||||
|
+namespace internal {
|
||||||
|
+
|
||||||
|
+#if EIGEN_HAS_C99_MATH
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct lgamma_impl<hip_impl::Scalar_accessor<T, U, n>> : lgamma_impl<T> {};
|
||||||
|
+#endif
|
||||||
|
+
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct digamma_impl_maybe_poly<hip_impl::Scalar_accessor<T, U, n>>
|
||||||
|
+ : digamma_impl_maybe_poly<T> {};
|
||||||
|
+
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct digamma_impl<hip_impl::Scalar_accessor<T, U, n>> : digamma_impl<T> {};
|
||||||
|
+
|
||||||
|
+#if EIGEN_HAS_C99_MATH
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct erf_impl<hip_impl::Scalar_accessor<T, U, n>> : erf_impl<T> {};
|
||||||
|
+#endif // EIGEN_HAS_C99_MATH
|
||||||
|
+
|
||||||
|
+#if EIGEN_HAS_C99_MATH
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct erfc_impl<hip_impl::Scalar_accessor<T, U, n>> : erfc_impl<T> {};
|
||||||
|
+#endif // EIGEN_HAS_C99_MATH
|
||||||
|
+
|
||||||
|
+#if EIGEN_HAS_C99_MATH
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct ndtri_impl<hip_impl::Scalar_accessor<T, U, n>> : ndtri_impl<T> {};
|
||||||
|
+#endif // EIGEN_HAS_C99_MATH
|
||||||
|
+
|
||||||
|
+template <typename T, typename U, unsigned int n, IgammaComputationMode mode>
|
||||||
|
+struct igammac_cf_impl<hip_impl::Scalar_accessor<T, U, n>, mode>
|
||||||
|
+ : igammac_cf_impl<T, mode> {};
|
||||||
|
+
|
||||||
|
+template <typename T, typename U, unsigned int n, IgammaComputationMode mode>
|
||||||
|
+struct igamma_series_impl<hip_impl::Scalar_accessor<T, U, n>, mode>
|
||||||
|
+ : igamma_series_impl<T, mode> {};
|
||||||
|
+
|
||||||
|
+#if EIGEN_HAS_C99_MATH
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct igammac_impl<hip_impl::Scalar_accessor<T, U, n>> : igammac_impl<T> {};
|
||||||
|
+#endif // EIGEN_HAS_C99_MATH
|
||||||
|
+
|
||||||
|
+#if EIGEN_HAS_C99_MATH
|
||||||
|
+template <typename T, typename U, unsigned int n, IgammaComputationMode mode>
|
||||||
|
+struct igamma_generic_impl<hip_impl::Scalar_accessor<T, U, n>, mode>
|
||||||
|
+ : igamma_generic_impl<T, mode> {};
|
||||||
|
+#endif // EIGEN_HAS_C99_MATH
|
||||||
|
+
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct igamma_impl<hip_impl::Scalar_accessor<T, U, n>> : igamma_impl<T> {};
|
||||||
|
+
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct igamma_der_a_retval<hip_impl::Scalar_accessor<T, U, n>>
|
||||||
|
+ : igamma_der_a_retval<T> {};
|
||||||
|
+
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct igamma_der_a_impl<hip_impl::Scalar_accessor<T, U, n>>
|
||||||
|
+ : igamma_der_a_impl<T> {};
|
||||||
|
+
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct gamma_sample_der_alpha_retval<hip_impl::Scalar_accessor<T, U, n>>
|
||||||
|
+ : gamma_sample_der_alpha_retval<T> {};
|
||||||
|
+
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct gamma_sample_der_alpha_impl<hip_impl::Scalar_accessor<T, U, n>>
|
||||||
|
+ : gamma_sample_der_alpha_impl<T> {};
|
||||||
|
+
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct zeta_impl_series<hip_impl::Scalar_accessor<T, U, n>>
|
||||||
|
+ : zeta_impl_series<T> {};
|
||||||
|
+
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct zeta_impl<hip_impl::Scalar_accessor<T, U, n>> : zeta_impl<T> {};
|
||||||
|
+
|
||||||
|
+#if EIGEN_HAS_C99_MATH
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct polygamma_impl<hip_impl::Scalar_accessor<T, U, n>>
|
||||||
|
+ : polygamma_impl<T> {};
|
||||||
|
+#endif // EIGEN_HAS_C99_MATH
|
||||||
|
+
|
||||||
|
+#if EIGEN_HAS_C99_MATH
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct betainc_impl<hip_impl::Scalar_accessor<T, U, n>> : betainc_impl<T> {};
|
||||||
|
+
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct incbeta_cfe<hip_impl::Scalar_accessor<T, U, n>> : incbeta_cfe<T> {};
|
||||||
|
+
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct betainc_helper<hip_impl::Scalar_accessor<T, U, n>>
|
||||||
|
+ : betainc_helper<T> {};
|
||||||
|
+#else
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct betainc_impl<hip_impl::Scalar_accessor<T, U, n>> : betainc_impl<T> {};
|
||||||
|
+#endif // EIGEN_HAS_C99_MATH
|
||||||
|
+
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct bessel_i0e_impl<hip_impl::Scalar_accessor<T, U, n>> : bessel_i0e_impl<T> {};
|
||||||
|
+
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct bessel_i0_impl<hip_impl::Scalar_accessor<T, U, n>> : bessel_i0_impl<T> {};
|
||||||
|
+
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct bessel_i1e_impl<hip_impl::Scalar_accessor<T, U, n>> : bessel_i1e_impl<T> {};
|
||||||
|
+
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct bessel_i1_impl<hip_impl::Scalar_accessor<T, U, n>> : bessel_i1_impl<T> {};
|
||||||
|
+
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct bessel_k0e_impl<hip_impl::Scalar_accessor<T, U, n>> : bessel_k0e_impl<T> {};
|
||||||
|
+
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct bessel_k0_impl<hip_impl::Scalar_accessor<T, U, n>> : bessel_k0_impl<T> {};
|
||||||
|
+
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct bessel_k1e_impl<hip_impl::Scalar_accessor<T, U, n>> : bessel_k1e_impl<T> {};
|
||||||
|
+
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct bessel_k1_impl<hip_impl::Scalar_accessor<T, U, n>> : bessel_k1_impl<T> {};
|
||||||
|
+
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct bessel_j0_impl<hip_impl::Scalar_accessor<T, U, n>> : bessel_j0_impl<T> {};
|
||||||
|
+
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct bessel_y0_impl<hip_impl::Scalar_accessor<T, U, n>> : bessel_y0_impl<T> {};
|
||||||
|
+
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct bessel_j1_impl<hip_impl::Scalar_accessor<T, U, n>> : bessel_j1_impl<T> {};
|
||||||
|
+
|
||||||
|
+template <typename T, typename U, unsigned int n>
|
||||||
|
+struct bessel_y1_impl<hip_impl::Scalar_accessor<T, U, n>> : bessel_y1_impl<T> {};
|
||||||
|
+
|
||||||
|
+} // end namespace internal
|
||||||
|
+} // end namespace Eigen
|
||||||
|
+
|
||||||
|
+#endif // HIP_VECTOR_COMPATIBILITY_H
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user