[MLIR][KernelGen] Enable MLIR-generated `tf.Xlogy` kernels by default
PiperOrigin-RevId: 387063554
Change-Id: Iec7c6130c42d2b57206c84b34037ee6723663e0d
diff --git a/tensorflow/core/kernels/cwise_op_gpu_xlogy.cu.cc b/tensorflow/core/kernels/cwise_op_gpu_xlogy.cu.cc
index f1cff47..7b367f0 100644
--- a/tensorflow/core/kernels/cwise_op_gpu_xlogy.cu.cc
+++ b/tensorflow/core/kernels/cwise_op_gpu_xlogy.cu.cc
@@ -20,14 +20,10 @@
namespace tensorflow {
namespace functor {
-#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
-#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \
- !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED)
+#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED)
DEFINE_BINARY3(xlogy, Eigen::half, float, double);
-#endif
-#endif
-
DEFINE_BINARY2(xlogy, complex64, complex128);
+#endif
} // namespace functor
} // namespace tensorflow
diff --git a/tensorflow/core/kernels/cwise_op_xlogy.cc b/tensorflow/core/kernels/cwise_op_xlogy.cc
index f2067c8..04a1def 100644
--- a/tensorflow/core/kernels/cwise_op_xlogy.cc
+++ b/tensorflow/core/kernels/cwise_op_xlogy.cc
@@ -21,8 +21,7 @@
complex64, complex128);
#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
-#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \
- !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED)
+#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED)
REGISTER3(BinaryOp, GPU, "Xlogy", functor::xlogy, float, Eigen::half, double);
REGISTER2(BinaryOp, GPU, "Xlogy", functor::xlogy, complex64, complex128);
#endif
diff --git a/tensorflow/core/kernels/mlir_generated/BUILD b/tensorflow/core/kernels/mlir_generated/BUILD
index 7c8ccd7..c304120 100644
--- a/tensorflow/core/kernels/mlir_generated/BUILD
+++ b/tensorflow/core/kernels/mlir_generated/BUILD
@@ -261,9 +261,9 @@
"gpu_op_squared_difference.cc",
"gpu_op_sub.cc",
"gpu_op_xdivy.cc",
+ "gpu_op_xlogy.cc",
"gpu_op_zeta.cc",
]) + if_mlir_generated_experimental_kernels_enabled([
- "gpu_op_xlogy.cc",
"gpu_op_xlog1py.cc",
]),
tags = ["manual"],
@@ -298,10 +298,10 @@
":gpu_squared_difference_kernels",
":gpu_sub_kernels",
":gpu_xdivy_kernels",
+ ":gpu_xlogy_kernels",
":gpu_zeta_kernels",
"//third_party/eigen3",
]) + if_mlir_generated_experimental_kernels_enabled([
- ":gpu_xlogy_kernels",
":gpu_xlog1py_kernels",
]),
)
