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

Fix ODR failures in TensorRandom.

Browse Source 
(cherry picked from commit bded5028a5bd112181b94b2a246ac2c20e671c2f)
This commit is contained in:
Antonio Sanchez 2022-02-11 23:19:56 -08:00
parent 709d704819
commit 21e0ad056e
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
unsupported/Eigen/CXX11/src/Tensor/TensorRandom.h
View File

@ -14,7 +14,7 @@
namespace Eigen { namespace Eigen {
namespace internal { namespace internal {
EIGEN_DEVICE_FUNC uint64_t get_random_seed() { EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE uint64_t get_random_seed() {
#if defined(EIGEN_GPU_COMPILE_PHASE) #if defined(EIGEN_GPU_COMPILE_PHASE)
// We don't support 3d kernels since we currently only use 1 and // We don't support 3d kernels since we currently only use 1 and
// 2d kernels. // 2d kernels.
@ -27,7 +27,7 @@ EIGEN_DEVICE_FUNC uint64_t get_random_seed() {
#endif #endif
} }
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE unsigned PCG_XSH_RS_generator(uint64_t* state, uint64_t stream) { EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE unsigned PCG_XSH_RS_generator(uint64_t* state, uint64_t stream) {
// TODO: Unify with the implementation in the non blocking thread pool. // TODO: Unify with the implementation in the non blocking thread pool.
uint64_t current = *state; uint64_t current = *state;
// Update the internal state // Update the internal state
@ -36,7 +36,7 @@ static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE unsigned PCG_XSH_RS_generator(uint6
return static_cast<unsigned>((current ^ (current >> 22)) >> (22 + (current >> 61))); return static_cast<unsigned>((current ^ (current >> 22)) >> (22 + (current >> 61)));
} }
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE uint64_t PCG_XSH_RS_state(uint64_t seed) { EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE uint64_t PCG_XSH_RS_state(uint64_t seed) {
seed = seed ? seed : get_random_seed(); seed = seed ? seed : get_random_seed();
return seed * 6364136223846793005ULL + 0xda3e39cb94b95bdbULL; return seed * 6364136223846793005ULL + 0xda3e39cb94b95bdbULL;
} }