diff options
| author |  A. Unique TensorFlower <nobody@tensorflow.org> | 2016-03-08 16:04:53 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-03-08 17:21:02 -0800 |
| commit | e1ac9aea1ca0a5aac83fa5b0f1e0929e412e44bd (patch) | |
| tree | f50879ca3c3efbbac47a708aab1748d0f08ead98 /tensorflow/core/lib/random | |
| parent | 6c576f017156cc0bf50e2dd2cd42b2ab8db9bffe (diff) | |
Use __host__ __device__ attributes during host and device-side CUDA compilation.
Change: 116708519
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 5fc2d5d20d..dc8de09d2c 100644 --- a/tensorflow/core/lib/random/philox_random.h +++ b/tensorflow/core/lib/random/philox_random.h @@ -25,7 +25,7 @@ limitations under the License. #include "tensorflow/core/platform/types.h" // Function qualifiers that need to work on both CPU and GPU. -#ifdef __CUDA_ARCH__ +#if defined(__CUDACC__) // For nvcc. #define PHILOX_DEVICE_FUNC __host__ __device__ #define PHILOX_INLINE __inline__ |