enable cwise op zeta for ROCm

tensorflow · Jun 3, 2019 · 119d9a1 · 119d9a1
1 parent 83a0c0e
commit 119d9a1
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tensorflow/core/kernels/cwise_op_zeta.cc b/tensorflow/core/kernels/cwise_op_zeta.cc
@@ -19,10 +19,8 @@ namespace tensorflow {
 REGISTER2(BinaryOp, CPU, "Zeta", functor::zeta, float, double);
 REGISTER2(BinaryOp, CPU, "Polygamma", functor::polygamma, float, double);
 
-#if GOOGLE_CUDA
-REGISTER2(BinaryOp, GPU, "Zeta", functor::zeta, float, double);
-#endif
 #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
+REGISTER2(BinaryOp, GPU, "Zeta", functor::zeta, float, double);
 REGISTER2(BinaryOp, GPU, "Polygamma", functor::polygamma, float, double);
 #endif
 }  // namespace tensorflow