Enable mlir generated GPU kernels by default for cuda builds.

PiperOrigin-RevId: 323742641
Change-Id: I97f7cc8743eafe85fbbb26672131df0cb4388c52

.bazelrc

@@ -173,11 +173,6 @@ build:using_cuda --define=using_cuda=true
 build:using_cuda --action_env TF_NEED_CUDA=1
 build:using_cuda --crosstool_top=@local_config_cuda//crosstool:toolchain
 
-# Enable the mlir generated GPU kernels only for cuda builds.
-build --define=tensorflow_enable_mlir_generated_kernels=0
-# This is a more specific option, so it takes precedence over the line above for cuda builds.
-build:using_cuda --define=tensorflow_enable_mlir_generated_kernels=1
-
 # This config refers to building CUDA op kernels with nvcc.
 build:cuda --config=using_cuda
 build:cuda --define=using_cuda_nvcc=true

tensorflow/core/kernels/mlir_generated/BUILD

@@ -18,9 +18,9 @@ package(
 )
 
 config_setting(
-    name = "mlir_generated_gpu_kernels_disabled",
+    name = "mlir_generated_gpu_kernels_enabled",
     define_values = {
-        "tensorflow_enable_mlir_generated_gpu_kernels": "0",
+        "tensorflow_enable_mlir_generated_gpu_kernels": "1",
     },
 )
 

tensorflow/core/kernels/mlir_generated/build_defs.bzl

@@ -4,8 +4,8 @@ load("@local_config_cuda//cuda:build_defs.bzl", "cuda_gpu_architectures", "if_cu
 
 def if_mlir_generated_gpu_kernels_enabled(if_true, if_false = []):
     return select({
-        "//tensorflow/core/kernels/mlir_generated:mlir_generated_gpu_kernels_disabled": if_false,
-        "//conditions:default": if_true,
+        "//tensorflow/core/kernels/mlir_generated:mlir_generated_gpu_kernels_enabled": if_true,
+        "//conditions:default": if_false,
     })
 
 def _lookup_file(filegroup, path):