From b98c9eb073d013211c50f4931e898b82e93ced32 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Thu, 19 Dec 2019 11:05:25 -0800 Subject: [PATCH] Add _USE_MATH_DEFINES to cwise_ops_gpu_common.cu.h to fix Windows GPU build. PiperOrigin-RevId: 286425447 Change-Id: I041e9036dc882345f4eeabd7966aa4b6dc6ad6ee --- tensorflow/core/kernels/cwise_ops_gpu_common.cu.h | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tensorflow/core/kernels/cwise_ops_gpu_common.cu.h b/tensorflow/core/kernels/cwise_ops_gpu_common.cu.h index cb042fb7a55..ecc58da315f 100644 --- a/tensorflow/core/kernels/cwise_ops_gpu_common.cu.h +++ b/tensorflow/core/kernels/cwise_ops_gpu_common.cu.h @@ -20,10 +20,11 @@ limitations under the License. #ifndef TENSORFLOW_CORE_KERNELS_CWISE_OPS_GPU_COMMON_CU_H_ #define TENSORFLOW_CORE_KERNELS_CWISE_OPS_GPU_COMMON_CU_H_ -#define EIGEN_USE_GPU - +#define _USE_MATH_DEFINES +#include #include +#define EIGEN_USE_GPU #include "tensorflow/core/framework/tensor_types.h" #include "tensorflow/core/kernels/cwise_ops.h" #include "tensorflow/core/platform/types.h"