experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add _USE_MATH_DEFINES to cwise_ops_gpu_common.cu.h to fix Windows GPU build.

Browse Source 
PiperOrigin-RevId: 286425447
Change-Id: I041e9036dc882345f4eeabd7966aa4b6dc6ad6ee
This commit is contained in:
A. Unique TensorFlower 2019-12-19 11:05:25 -08:00 committed by TensorFlower Gardener
parent bb63451891
commit b98c9eb073
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/core/kernels/cwise_ops_gpu_common.cu.h
View File

@ -20,10 +20,11 @@ limitations under the License.
#ifndef TENSORFLOW_CORE_KERNELS_CWISE_OPS_GPU_COMMON_CU_H_ #ifndef TENSORFLOW_CORE_KERNELS_CWISE_OPS_GPU_COMMON_CU_H_
#define 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 <cmath>
#include <complex> #include <complex>
#define EIGEN_USE_GPU
#include "tensorflow/core/framework/tensor_types.h" #include "tensorflow/core/framework/tensor_types.h"
#include "tensorflow/core/kernels/cwise_ops.h" #include "tensorflow/core/kernels/cwise_ops.h"
#include "tensorflow/core/platform/types.h" #include "tensorflow/core/platform/types.h"