experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request from ROCmSoftwarePlatform:google-upstream-eigen-patch

Browse Source 
PiperOrigin-RevId: 285937983
Change-Id: I7961e539f271f97036b32f58260d1973cb1728c6
This commit is contained in:
TensorFlower Gardener 2019-12-17 01:56:18 -08:00
commit 1768c8f2fa
1 changed files with 158 additions and 1 deletions

159
third_party/eigen3/gpu_packet_math.patch vendored
View File

@ -22,4 +22,161 @@
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