diff options
author | A. Unique TensorFlower <nobody@tensorflow.org> | 2016-05-27 15:26:11 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-05-27 16:36:45 -0700 |
commit | 895a0a423bca9118faa66fb144d434e5c06751cc (patch) | |
tree | 6bda493292785f0d722a04a70f30c75fb3175993 /tensorflow/core/lib/random | |
parent | 47b21485352878e080909117aa783d2051bae783 (diff) |
Make PhiloxRandom use __umulhi on all CUDA compiles.
Change: 123468391
Diffstat (limited to 'tensorflow/core/lib/random')
-rw-r--r-- | tensorflow/core/lib/random/philox_random.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/lib/random/philox_random.h b/tensorflow/core/lib/random/philox_random.h index 3e53eeae33..9f67e64a06 100644 --- a/tensorflow/core/lib/random/philox_random.h +++ b/tensorflow/core/lib/random/philox_random.h @@ -204,7 +204,7 @@ class PhiloxRandom { PHILOX_DEVICE_INLINE static void MultiplyHighLow(uint32 a, uint32 b, uint32* result_low, uint32* result_high) { -#ifndef __GCUDACC__ +#ifndef __CUDA_ARCH__ const uint64 product = static_cast<uint64>(a) * b; *result_low = static_cast<uint32>(product); *result_high = static_cast<uint32>(product >> 32); |