Skip to content

Commit d542486

Browse files
WindQAQfacaiy
authored andcommitted
Fix GOOGLE_CUDA=1 flag (#460)
1 parent bbec769 commit d542486

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

tensorflow_addons/custom_ops/activations/cc/kernels/gelu_op.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ using CPUDevice = Eigen::ThreadPoolDevice;
3636
TF_CALL_GPU_NUMBER_TYPES(REGISTER_GELU_KERNELS);
3737
#undef REGISTER_GELU_KERNELS
3838

39-
#ifdef GOOGLE_CUDA
39+
#if GOOGLE_CUDA
4040

4141
using GPUDevice = Eigen::GpuDevice;
4242

tensorflow_addons/custom_ops/layers/cc/kernels/correlation_cost_op.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -331,7 +331,7 @@ TF_CALL_float(REGISTER_CORRELATIONCOST_OP_CPU);
331331
#undef REGISTER_CORRELATIONCOST_OP_CPU
332332

333333
// Register the GPU kernels.
334-
#ifdef GOOGLE_CUDA
334+
#if GOOGLE_CUDA
335335

336336
#define REGISTER_CORRELATIONCOST_OP_GPU(T) \
337337
REGISTER_KERNEL_BUILDER( \

0 commit comments

Comments
 (0)