Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
wux_labs
Tensorflow
提交
9837eceb
T
Tensorflow
项目概览
wux_labs
/
Tensorflow
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
T
Tensorflow
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
9837eceb
编写于
12月 20, 2019
作者:
G
Goldie Gadde
提交者:
GitHub
12月 20, 2019
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #35238 from ROCmSoftwarePlatform/r2.1-eigen-patch
[ROCm][r2.1] eigen patch needed for HIP header changes
上级
c23091df
2b8a5652
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
158 addition
and
1 deletion
+158
-1
third_party/eigen3/gpu_packet_math.patch
third_party/eigen3/gpu_packet_math.patch
+158
-1
未找到文件。
third_party/eigen3/gpu_packet_math.patch
浏览文件 @
9837eceb
...
...
@@ -22,4 +22,161 @@
return res;
}
};
\ No newline at end of file
--- 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
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录