Skip to content

Commit 550dc59

Browse files
authored
Merge pull request #1 from minty99/restrict
Fix accidentally deleted #if line
2 parents a31b245 + a317cb2 commit 550dc59

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorflow/core/kernels/relu_op_gpu.cu.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@ struct ReluGrad<Device, Eigen::half> {
118118
};
119119
#endif // GOOGLE_CUDA
120120

121-
121+
#if GOOGLE_CUDA
122122
__global__ void Relu_int8x4_kernel(int vect_count, const int32* __restrict__ input,
123123
int32* __restrict__ output) {
124124
CUDA_1D_KERNEL_LOOP(index, vect_count) {

0 commit comments

Comments
 (0)