From bff7bf1e48d40a5cc95c291fea042c93e5c20af1 Mon Sep 17 00:00:00 2001 From: Alexander Belyaev Date: Wed, 3 Feb 2021 04:39:20 -0800 Subject: [PATCH] [KERNEL_GEN] Rename MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED -> MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED. The plan is to use this flag for experimental CPU kernels as well. PiperOrigin-RevId: 355369970 Change-Id: I5bdb413176c3aa7cb753b3aa554046a05eb799f3 --- tensorflow/core/kernels/BUILD | 2 +- tensorflow/core/kernels/cwise_op_acos.cc | 2 +- tensorflow/core/kernels/cwise_op_acosh.cc | 2 +- tensorflow/core/kernels/cwise_op_add_1.cc | 2 +- tensorflow/core/kernels/cwise_op_add_2.cc | 2 +- tensorflow/core/kernels/cwise_op_arg.cc | 2 +- tensorflow/core/kernels/cwise_op_asin.cc | 2 +- tensorflow/core/kernels/cwise_op_asinh.cc | 2 +- tensorflow/core/kernels/cwise_op_atan2.cc | 2 +- tensorflow/core/kernels/cwise_op_bitwise_and.cc | 4 ++-- tensorflow/core/kernels/cwise_op_bitwise_or.cc | 4 ++-- tensorflow/core/kernels/cwise_op_bitwise_xor.cc | 4 ++-- tensorflow/core/kernels/cwise_op_digamma.cc | 2 +- tensorflow/core/kernels/cwise_op_div.cc | 2 +- tensorflow/core/kernels/cwise_op_equal_to_1.cc | 2 +- tensorflow/core/kernels/cwise_op_equal_to_2.cc | 2 +- tensorflow/core/kernels/cwise_op_exp.cc | 2 +- tensorflow/core/kernels/cwise_op_expm1.cc | 2 +- tensorflow/core/kernels/cwise_op_floor_div.cc | 2 +- tensorflow/core/kernels/cwise_op_gpu_acos.cu.cc | 2 +- tensorflow/core/kernels/cwise_op_gpu_acosh.cu.cc | 2 +- tensorflow/core/kernels/cwise_op_gpu_arg.cu.cc | 2 +- tensorflow/core/kernels/cwise_op_gpu_left_shift.cu.cc | 2 +- tensorflow/core/kernels/cwise_op_gpu_maximum.cu.cc | 2 +- tensorflow/core/kernels/cwise_op_gpu_minimum.cu.cc | 2 +- tensorflow/core/kernels/cwise_op_gpu_pow.cu.cc | 2 +- tensorflow/core/kernels/cwise_op_gpu_right_shift.cu.cc | 2 +- tensorflow/core/kernels/cwise_op_gpu_squared_difference.cu.cc | 2 +- tensorflow/core/kernels/cwise_op_greater.cc | 2 +- tensorflow/core/kernels/cwise_op_greater_equal.cc | 2 +- tensorflow/core/kernels/cwise_op_left_shift.cc | 2 +- tensorflow/core/kernels/cwise_op_less.cc | 2 +- tensorflow/core/kernels/cwise_op_less_equal.cc | 2 +- tensorflow/core/kernels/cwise_op_lgamma.cc | 2 +- tensorflow/core/kernels/cwise_op_logical_and.cc | 4 ++-- tensorflow/core/kernels/cwise_op_logical_or.cc | 4 ++-- tensorflow/core/kernels/cwise_op_maximum.cc | 2 +- tensorflow/core/kernels/cwise_op_minimum.cc | 2 +- tensorflow/core/kernels/cwise_op_mul_1.cc | 2 +- tensorflow/core/kernels/cwise_op_mul_2.cc | 2 +- tensorflow/core/kernels/cwise_op_not_equal_to_1.cc | 2 +- tensorflow/core/kernels/cwise_op_not_equal_to_2.cc | 2 +- tensorflow/core/kernels/cwise_op_pow.cc | 2 +- tensorflow/core/kernels/cwise_op_right_shift.cc | 2 +- tensorflow/core/kernels/cwise_op_sign.cc | 2 +- tensorflow/core/kernels/cwise_op_sin.cc | 2 +- tensorflow/core/kernels/cwise_op_sinh.cc | 2 +- tensorflow/core/kernels/cwise_op_squared_difference.cc | 2 +- tensorflow/core/kernels/cwise_op_sub.cc | 2 +- 49 files changed, 54 insertions(+), 54 deletions(-) diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD index 471008157af..e2d5d2c7ca8 100644 --- a/tensorflow/core/kernels/BUILD +++ b/tensorflow/core/kernels/BUILD @@ -3347,7 +3347,7 @@ tf_kernel_library( tf_kernel_library( name = "cwise_op", copts = if_mlir_generated_gpu_kernels_enabled(if_true = ["-DMLIR_GENERATED_GPU_KERNELS_ENABLED=1"]) + - if_mlir_experimental_kernels_enabled(if_true = ["-DMLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED=1"]), + if_mlir_experimental_kernels_enabled(if_true = ["-DMLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED=1"]), prefix = "cwise_op", deps = MATH_DEPS + if_mlir_generated_gpu_kernels_enabled(if_true = ["//tensorflow/core/kernels/mlir_generated:cwise_op"]), ) diff --git a/tensorflow/core/kernels/cwise_op_acos.cc b/tensorflow/core/kernels/cwise_op_acos.cc index 9e571347290..2e75adf0c95 100644 --- a/tensorflow/core/kernels/cwise_op_acos.cc +++ b/tensorflow/core/kernels/cwise_op_acos.cc @@ -20,7 +20,7 @@ REGISTER2(UnaryOp, CPU, "Acos", functor::acos, float, double); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER2(UnaryOp, GPU, "Acos", functor::acos, float, double); #endif #endif diff --git a/tensorflow/core/kernels/cwise_op_acosh.cc b/tensorflow/core/kernels/cwise_op_acosh.cc index 5682f8d6115..6b0a85b3cfe 100644 --- a/tensorflow/core/kernels/cwise_op_acosh.cc +++ b/tensorflow/core/kernels/cwise_op_acosh.cc @@ -23,7 +23,7 @@ REGISTER4(UnaryOp, CPU, "Acosh", functor::acosh, float, double, complex64, #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER2(UnaryOp, GPU, "Acosh", functor::acosh, float, double); #endif #endif diff --git a/tensorflow/core/kernels/cwise_op_add_1.cc b/tensorflow/core/kernels/cwise_op_add_1.cc index e5ef14aa5cd..f8547658a12 100644 --- a/tensorflow/core/kernels/cwise_op_add_1.cc +++ b/tensorflow/core/kernels/cwise_op_add_1.cc @@ -25,7 +25,7 @@ REGISTER6(BinaryOp, CPU, "AddV2", functor::add, float, Eigen::half, double, REGISTER3(BinaryOp, GPU, "Add", functor::add, float, Eigen::half, double); #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER3(BinaryOp, GPU, "AddV2", functor::add, float, Eigen::half, double); #endif diff --git a/tensorflow/core/kernels/cwise_op_add_2.cc b/tensorflow/core/kernels/cwise_op_add_2.cc index 8f1890c3458..f1d0fbdfe52 100644 --- a/tensorflow/core/kernels/cwise_op_add_2.cc +++ b/tensorflow/core/kernels/cwise_op_add_2.cc @@ -33,7 +33,7 @@ REGISTER6(BinaryOp, GPU, "Add", functor::add, uint8, uint16, uint64, int64, complex64, complex128); #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER7(BinaryOp, GPU, "AddV2", functor::add, uint8, uint16, uint32, uint64, int64, complex64, complex128); #else diff --git a/tensorflow/core/kernels/cwise_op_arg.cc b/tensorflow/core/kernels/cwise_op_arg.cc index 9a8122f1081..b847bd429ce 100644 --- a/tensorflow/core/kernels/cwise_op_arg.cc +++ b/tensorflow/core/kernels/cwise_op_arg.cc @@ -28,7 +28,7 @@ REGISTER_COMPLEX(CPU, double, complex128); #if GOOGLE_CUDA #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER_COMPLEX(GPU, float, complex64); REGISTER_COMPLEX(GPU, double, complex128); #endif diff --git a/tensorflow/core/kernels/cwise_op_asin.cc b/tensorflow/core/kernels/cwise_op_asin.cc index d9b41abb8e7..db6e21bfffe 100644 --- a/tensorflow/core/kernels/cwise_op_asin.cc +++ b/tensorflow/core/kernels/cwise_op_asin.cc @@ -20,7 +20,7 @@ REGISTER2(UnaryOp, CPU, "Asin", functor::asin, float, double); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER2(UnaryOp, GPU, "Asin", functor::asin, float, double); #endif #endif diff --git a/tensorflow/core/kernels/cwise_op_asinh.cc b/tensorflow/core/kernels/cwise_op_asinh.cc index 6d162c93b42..28356935905 100644 --- a/tensorflow/core/kernels/cwise_op_asinh.cc +++ b/tensorflow/core/kernels/cwise_op_asinh.cc @@ -22,7 +22,7 @@ REGISTER4(UnaryOp, CPU, "Asinh", functor::asinh, float, double, complex64, #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER2(UnaryOp, GPU, "Asinh", functor::asinh, float, double); #endif #endif diff --git a/tensorflow/core/kernels/cwise_op_atan2.cc b/tensorflow/core/kernels/cwise_op_atan2.cc index ea5589a6869..815e498692b 100644 --- a/tensorflow/core/kernels/cwise_op_atan2.cc +++ b/tensorflow/core/kernels/cwise_op_atan2.cc @@ -21,7 +21,7 @@ REGISTER2(BinaryOp, CPU, "Atan2", functor::atan2, float, double); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER2(BinaryOp, GPU, "Atan2", functor::atan2, float, double); #endif #endif diff --git a/tensorflow/core/kernels/cwise_op_bitwise_and.cc b/tensorflow/core/kernels/cwise_op_bitwise_and.cc index 63d47d1c1f4..5858b1d768d 100644 --- a/tensorflow/core/kernels/cwise_op_bitwise_and.cc +++ b/tensorflow/core/kernels/cwise_op_bitwise_and.cc @@ -22,7 +22,7 @@ REGISTER8(BinaryOp, CPU, "BitwiseAnd", functor::bitwise_and, int8, int16, int32, #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER8(BinaryOp, GPU, "BitwiseAnd", functor::bitwise_and, int8, int16, int32, int64, uint8, uint16, uint32, uint64); #else @@ -30,7 +30,7 @@ REGISTER8(BinaryOp, GPU, "BitwiseAnd", functor::bitwise_and, int8, int16, int32, REGISTER4(BinaryOp, GPU, "BitwiseAnd", functor::bitwise_and, uint8, uint16, uint32, uint64); #endif // !MLIR_GENERATED_GPU_KERNELS_ENABLED || - // !MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED + // !MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED #endif // GOOGLE_CUDA || TENSORFLOW_USE_ROCM } // namespace tensorflow diff --git a/tensorflow/core/kernels/cwise_op_bitwise_or.cc b/tensorflow/core/kernels/cwise_op_bitwise_or.cc index 7457845fcc2..108164567cf 100644 --- a/tensorflow/core/kernels/cwise_op_bitwise_or.cc +++ b/tensorflow/core/kernels/cwise_op_bitwise_or.cc @@ -22,7 +22,7 @@ REGISTER8(BinaryOp, CPU, "BitwiseOr", functor::bitwise_or, int8, int16, int32, #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER8(BinaryOp, GPU, "BitwiseOr", functor::bitwise_or, int8, int16, int32, int64, uint8, uint16, uint32, uint64); #else @@ -30,7 +30,7 @@ REGISTER8(BinaryOp, GPU, "BitwiseOr", functor::bitwise_or, int8, int16, int32, REGISTER4(BinaryOp, GPU, "BitwiseOr", functor::bitwise_or, uint8, uint16, uint32, uint64); #endif // !MLIR_GENERATED_GPU_KERNELS_ENABLED || - // !MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED + // !MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED #endif // GOOGLE_CUDA || TENSORFLOW_USE_ROCM } // namespace tensorflow diff --git a/tensorflow/core/kernels/cwise_op_bitwise_xor.cc b/tensorflow/core/kernels/cwise_op_bitwise_xor.cc index 0569bfed1e2..629bb00ea4a 100644 --- a/tensorflow/core/kernels/cwise_op_bitwise_xor.cc +++ b/tensorflow/core/kernels/cwise_op_bitwise_xor.cc @@ -22,7 +22,7 @@ REGISTER8(BinaryOp, CPU, "BitwiseXor", functor::bitwise_xor, int8, int16, int32, #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER8(BinaryOp, GPU, "BitwiseXor", functor::bitwise_xor, int8, int16, int32, int64, uint8, uint16, uint32, uint64); #else @@ -30,7 +30,7 @@ REGISTER8(BinaryOp, GPU, "BitwiseXor", functor::bitwise_xor, int8, int16, int32, REGISTER4(BinaryOp, GPU, "BitwiseXor", functor::bitwise_xor, uint8, uint16, uint32, uint64); #endif // !MLIR_GENERATED_GPU_KERNELS_ENABLED || - // !MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED + // !MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED #endif // GOOGLE_CUDA || TENSORFLOW_USE_ROCM } // namespace tensorflow diff --git a/tensorflow/core/kernels/cwise_op_digamma.cc b/tensorflow/core/kernels/cwise_op_digamma.cc index f01f7b9d24d..879c3f2938b 100644 --- a/tensorflow/core/kernels/cwise_op_digamma.cc +++ b/tensorflow/core/kernels/cwise_op_digamma.cc @@ -22,7 +22,7 @@ REGISTER3(UnaryOp, CPU, "Digamma", functor::digamma, float, Eigen::half, #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER3(UnaryOp, GPU, "Digamma", functor::digamma, float, Eigen::half, double); #endif diff --git a/tensorflow/core/kernels/cwise_op_div.cc b/tensorflow/core/kernels/cwise_op_div.cc index 5f45c922e74..430eacf9518 100644 --- a/tensorflow/core/kernels/cwise_op_div.cc +++ b/tensorflow/core/kernels/cwise_op_div.cc @@ -30,7 +30,7 @@ REGISTER5(BinaryOp, CPU, "DivNoNan", functor::div_no_nan, Eigen::half, float, #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM // ROCM TODO: re-enable complex64 / complex128 after compiler fix #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER9(BinaryOp, GPU, "Div", functor::div, float, Eigen::half, double, uint8, uint16, int16, int64, complex64, complex128); REGISTER5(BinaryOp, GPU, "RealDiv", functor::div, float, Eigen::half, double, diff --git a/tensorflow/core/kernels/cwise_op_equal_to_1.cc b/tensorflow/core/kernels/cwise_op_equal_to_1.cc index b0dfe945605..8b42e975ac8 100644 --- a/tensorflow/core/kernels/cwise_op_equal_to_1.cc +++ b/tensorflow/core/kernels/cwise_op_equal_to_1.cc @@ -28,7 +28,7 @@ REGISTER_KERNEL_BUILDER( ApproximateEqualOp); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER4(BinaryOp, GPU, "Equal", functor::equal_to, float, Eigen::half, double, uint8); #else diff --git a/tensorflow/core/kernels/cwise_op_equal_to_2.cc b/tensorflow/core/kernels/cwise_op_equal_to_2.cc index 95eb576ba07..2f60f8e74b7 100644 --- a/tensorflow/core/kernels/cwise_op_equal_to_2.cc +++ b/tensorflow/core/kernels/cwise_op_equal_to_2.cc @@ -26,7 +26,7 @@ REGISTER6(BinaryOp, CPU, "Equal", functor::equal_to, int32, int64, complex64, complex128, tstring, bool); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER6(BinaryOp, GPU, "Equal", functor::equal_to, int8, int16, int64, complex64, complex128, bool); #else diff --git a/tensorflow/core/kernels/cwise_op_exp.cc b/tensorflow/core/kernels/cwise_op_exp.cc index 71833e623b6..47298e4de3a 100644 --- a/tensorflow/core/kernels/cwise_op_exp.cc +++ b/tensorflow/core/kernels/cwise_op_exp.cc @@ -21,7 +21,7 @@ REGISTER6(UnaryOp, CPU, "Exp", functor::exp, float, Eigen::half, bfloat16, #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER5(UnaryOp, GPU, "Exp", functor::exp, float, Eigen::half, double, complex64, complex128); #endif diff --git a/tensorflow/core/kernels/cwise_op_expm1.cc b/tensorflow/core/kernels/cwise_op_expm1.cc index d14939b884d..6bc8fd2ea5d 100644 --- a/tensorflow/core/kernels/cwise_op_expm1.cc +++ b/tensorflow/core/kernels/cwise_op_expm1.cc @@ -20,7 +20,7 @@ REGISTER6(UnaryOp, CPU, "Expm1", functor::expm1, float, Eigen::half, bfloat16, double, complex64, complex128); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER3(UnaryOp, GPU, "Expm1", functor::expm1, float, Eigen::half, double); #endif #endif diff --git a/tensorflow/core/kernels/cwise_op_floor_div.cc b/tensorflow/core/kernels/cwise_op_floor_div.cc index 4e12829d3ee..0295fbaa247 100644 --- a/tensorflow/core/kernels/cwise_op_floor_div.cc +++ b/tensorflow/core/kernels/cwise_op_floor_div.cc @@ -25,7 +25,7 @@ REGISTER4(BinaryOp, CPU, "FloorDiv", functor::floor_div_real, float, REGISTER4(BinaryOp, GPU, "FloorDiv", functor::floor_div, uint8, uint16, int16, int64); #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER3(BinaryOp, GPU, "FloorDiv", functor::floor_div_real, float, Eigen::half, double); #endif diff --git a/tensorflow/core/kernels/cwise_op_gpu_acos.cu.cc b/tensorflow/core/kernels/cwise_op_gpu_acos.cu.cc index 7592446b0df..032cf8b6a6f 100644 --- a/tensorflow/core/kernels/cwise_op_gpu_acos.cu.cc +++ b/tensorflow/core/kernels/cwise_op_gpu_acos.cu.cc @@ -20,7 +20,7 @@ limitations under the License. namespace tensorflow { namespace functor { #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) DEFINE_UNARY2(acos, float, double); #endif } // namespace functor diff --git a/tensorflow/core/kernels/cwise_op_gpu_acosh.cu.cc b/tensorflow/core/kernels/cwise_op_gpu_acosh.cu.cc index ca8166ce66b..6321bfc4bd2 100644 --- a/tensorflow/core/kernels/cwise_op_gpu_acosh.cu.cc +++ b/tensorflow/core/kernels/cwise_op_gpu_acosh.cu.cc @@ -21,7 +21,7 @@ limitations under the License. namespace tensorflow { namespace functor { #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) DEFINE_UNARY2(acosh, float, double); #endif } // namespace functor diff --git a/tensorflow/core/kernels/cwise_op_gpu_arg.cu.cc b/tensorflow/core/kernels/cwise_op_gpu_arg.cu.cc index 1291c9c5b78..e84f1573f07 100644 --- a/tensorflow/core/kernels/cwise_op_gpu_arg.cu.cc +++ b/tensorflow/core/kernels/cwise_op_gpu_arg.cu.cc @@ -21,7 +21,7 @@ namespace tensorflow { namespace functor { #if GOOGLE_CUDA #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) DEFINE_UNARY2(get_angle, complex64, complex128); #endif #endif diff --git a/tensorflow/core/kernels/cwise_op_gpu_left_shift.cu.cc b/tensorflow/core/kernels/cwise_op_gpu_left_shift.cu.cc index 9305a638848..b548cfa780c 100644 --- a/tensorflow/core/kernels/cwise_op_gpu_left_shift.cu.cc +++ b/tensorflow/core/kernels/cwise_op_gpu_left_shift.cu.cc @@ -20,7 +20,7 @@ limitations under the License. namespace tensorflow { namespace functor { #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) DEFINE_BINARY8(left_shift, int8, int16, int32, int64, uint8, uint16, uint32, uint64); #else diff --git a/tensorflow/core/kernels/cwise_op_gpu_maximum.cu.cc b/tensorflow/core/kernels/cwise_op_gpu_maximum.cu.cc index 5d529efff88..0a0fe73a870 100644 --- a/tensorflow/core/kernels/cwise_op_gpu_maximum.cu.cc +++ b/tensorflow/core/kernels/cwise_op_gpu_maximum.cu.cc @@ -20,7 +20,7 @@ limitations under the License. namespace tensorflow { namespace functor { #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) DEFINE_BINARY6(maximum, Eigen::half, float, double, uint8, int16, int64); #endif } // namespace functor diff --git a/tensorflow/core/kernels/cwise_op_gpu_minimum.cu.cc b/tensorflow/core/kernels/cwise_op_gpu_minimum.cu.cc index f033ca1ae20..c085a0daf57 100644 --- a/tensorflow/core/kernels/cwise_op_gpu_minimum.cu.cc +++ b/tensorflow/core/kernels/cwise_op_gpu_minimum.cu.cc @@ -20,7 +20,7 @@ limitations under the License. namespace tensorflow { namespace functor { #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) DEFINE_BINARY6(minimum, Eigen::half, float, double, uint8, int16, int64); #endif } // namespace functor diff --git a/tensorflow/core/kernels/cwise_op_gpu_pow.cu.cc b/tensorflow/core/kernels/cwise_op_gpu_pow.cu.cc index 1a4acb78f69..b309b33efbd 100644 --- a/tensorflow/core/kernels/cwise_op_gpu_pow.cu.cc +++ b/tensorflow/core/kernels/cwise_op_gpu_pow.cu.cc @@ -20,7 +20,7 @@ limitations under the License. namespace tensorflow { namespace functor { #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) DEFINE_BINARY3(pow, Eigen::half, float, double); DEFINE_BINARY1(safe_pow_ignore_error, int64); #endif diff --git a/tensorflow/core/kernels/cwise_op_gpu_right_shift.cu.cc b/tensorflow/core/kernels/cwise_op_gpu_right_shift.cu.cc index 6c02a6f99e5..5cc62807c96 100644 --- a/tensorflow/core/kernels/cwise_op_gpu_right_shift.cu.cc +++ b/tensorflow/core/kernels/cwise_op_gpu_right_shift.cu.cc @@ -20,7 +20,7 @@ limitations under the License. namespace tensorflow { namespace functor { #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) DEFINE_BINARY8(right_shift, int8, int16, int32, int64, uint8, uint16, uint32, uint64); #else diff --git a/tensorflow/core/kernels/cwise_op_gpu_squared_difference.cu.cc b/tensorflow/core/kernels/cwise_op_gpu_squared_difference.cu.cc index 76269e888f6..1e83c3e124e 100644 --- a/tensorflow/core/kernels/cwise_op_gpu_squared_difference.cu.cc +++ b/tensorflow/core/kernels/cwise_op_gpu_squared_difference.cu.cc @@ -20,7 +20,7 @@ limitations under the License. namespace tensorflow { namespace functor { #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) DEFINE_BINARY4(squared_difference, float, Eigen::half, double, int64); #endif } // namespace functor diff --git a/tensorflow/core/kernels/cwise_op_greater.cc b/tensorflow/core/kernels/cwise_op_greater.cc index 5f85bceb597..fac0ed8ce0d 100644 --- a/tensorflow/core/kernels/cwise_op_greater.cc +++ b/tensorflow/core/kernels/cwise_op_greater.cc @@ -21,7 +21,7 @@ REGISTER9(BinaryOp, CPU, "Greater", functor::greater, float, Eigen::half, REGISTER3(BinaryOp, CPU, "Greater", functor::greater, int8, int16, bfloat16); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER6(BinaryOp, GPU, "Greater", functor::greater, float, Eigen::half, double, int8, int16, int64); REGISTER4(BinaryOp, GPU, "Greater", functor::greater, uint8, uint16, uint32, diff --git a/tensorflow/core/kernels/cwise_op_greater_equal.cc b/tensorflow/core/kernels/cwise_op_greater_equal.cc index 2dd4a10da45..6d1b21b6835 100644 --- a/tensorflow/core/kernels/cwise_op_greater_equal.cc +++ b/tensorflow/core/kernels/cwise_op_greater_equal.cc @@ -22,7 +22,7 @@ REGISTER3(BinaryOp, CPU, "GreaterEqual", functor::greater_equal, int8, int16, bfloat16); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER9(BinaryOp, GPU, "GreaterEqual", functor::greater_equal, float, Eigen::half, double, int64, uint8, uint16, uint32, uint64, int8); REGISTER(BinaryOp, GPU, "GreaterEqual", functor::greater_equal, int16); diff --git a/tensorflow/core/kernels/cwise_op_left_shift.cc b/tensorflow/core/kernels/cwise_op_left_shift.cc index f43c46b0aef..6e05a512c31 100644 --- a/tensorflow/core/kernels/cwise_op_left_shift.cc +++ b/tensorflow/core/kernels/cwise_op_left_shift.cc @@ -22,7 +22,7 @@ REGISTER8(BinaryOp, CPU, "LeftShift", functor::left_shift, int8, int16, int32, #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER8(BinaryOp, GPU, "LeftShift", functor::left_shift, int8, int16, int32, int64, uint8, uint16, uint32, uint64); #else diff --git a/tensorflow/core/kernels/cwise_op_less.cc b/tensorflow/core/kernels/cwise_op_less.cc index 967145d300d..108d2e34a84 100644 --- a/tensorflow/core/kernels/cwise_op_less.cc +++ b/tensorflow/core/kernels/cwise_op_less.cc @@ -23,7 +23,7 @@ REGISTER7(BinaryOp, CPU, "Less", functor::less, uint8, uint16, uint32, uint64, #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER9(BinaryOp, GPU, "Less", functor::less, float, Eigen::half, double, int64, uint8, uint16, uint32, uint64, int8); REGISTER(BinaryOp, GPU, "Less", functor::less, int16); diff --git a/tensorflow/core/kernels/cwise_op_less_equal.cc b/tensorflow/core/kernels/cwise_op_less_equal.cc index e93368c00df..226ee9b2436 100644 --- a/tensorflow/core/kernels/cwise_op_less_equal.cc +++ b/tensorflow/core/kernels/cwise_op_less_equal.cc @@ -23,7 +23,7 @@ REGISTER7(BinaryOp, CPU, "LessEqual", functor::less_equal, int64, uint8, uint16, #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER9(BinaryOp, GPU, "LessEqual", functor::less_equal, float, Eigen::half, double, int64, uint8, uint16, uint32, uint64, int8); REGISTER(BinaryOp, GPU, "LessEqual", functor::less_equal, int16); diff --git a/tensorflow/core/kernels/cwise_op_lgamma.cc b/tensorflow/core/kernels/cwise_op_lgamma.cc index 86cc2479aa9..15734f2bb87 100644 --- a/tensorflow/core/kernels/cwise_op_lgamma.cc +++ b/tensorflow/core/kernels/cwise_op_lgamma.cc @@ -21,7 +21,7 @@ REGISTER3(UnaryOp, CPU, "Lgamma", functor::lgamma, float, Eigen::half, double); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER3(UnaryOp, GPU, "Lgamma", functor::lgamma, float, Eigen::half, double); #endif #endif diff --git a/tensorflow/core/kernels/cwise_op_logical_and.cc b/tensorflow/core/kernels/cwise_op_logical_and.cc index 8cd85176c1f..d814132e4cc 100644 --- a/tensorflow/core/kernels/cwise_op_logical_and.cc +++ b/tensorflow/core/kernels/cwise_op_logical_and.cc @@ -20,10 +20,10 @@ REGISTER_KERNEL_BUILDER(Name("LogicalAnd").Device(DEVICE_CPU), BinaryOp); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER_KERNEL_BUILDER(Name("LogicalAnd").Device(DEVICE_GPU), BinaryOp); #endif // !MLIR_GENERATED_GPU_KERNELS_ENABLED || - // !MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED + // !MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED #endif } // namespace tensorflow diff --git a/tensorflow/core/kernels/cwise_op_logical_or.cc b/tensorflow/core/kernels/cwise_op_logical_or.cc index 052e29ab88d..ad808f6f340 100644 --- a/tensorflow/core/kernels/cwise_op_logical_or.cc +++ b/tensorflow/core/kernels/cwise_op_logical_or.cc @@ -20,10 +20,10 @@ REGISTER_KERNEL_BUILDER(Name("LogicalOr").Device(DEVICE_CPU), BinaryOp); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER_KERNEL_BUILDER(Name("LogicalOr").Device(DEVICE_GPU), BinaryOp); #endif // !MLIR_GENERATED_GPU_KERNELS_ENABLED || - // !MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED + // !MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED #endif } // namespace tensorflow diff --git a/tensorflow/core/kernels/cwise_op_maximum.cc b/tensorflow/core/kernels/cwise_op_maximum.cc index fa7e57b3ac3..35422ba7980 100644 --- a/tensorflow/core/kernels/cwise_op_maximum.cc +++ b/tensorflow/core/kernels/cwise_op_maximum.cc @@ -20,7 +20,7 @@ REGISTER8(BinaryOp, CPU, "Maximum", functor::maximum, float, Eigen::half, bfloat16, double, uint8, int16, int32, int64); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER6(BinaryOp, GPU, "Maximum", functor::maximum, float, Eigen::half, double, uint8, int16, int64); #else diff --git a/tensorflow/core/kernels/cwise_op_minimum.cc b/tensorflow/core/kernels/cwise_op_minimum.cc index 16f5cae6b6a..384fbf4034a 100644 --- a/tensorflow/core/kernels/cwise_op_minimum.cc +++ b/tensorflow/core/kernels/cwise_op_minimum.cc @@ -20,7 +20,7 @@ REGISTER8(BinaryOp, CPU, "Minimum", functor::minimum, float, Eigen::half, bfloat16, double, uint8, int16, int32, int64); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER6(BinaryOp, GPU, "Minimum", functor::minimum, float, Eigen::half, double, uint8, int16, int64); #else diff --git a/tensorflow/core/kernels/cwise_op_mul_1.cc b/tensorflow/core/kernels/cwise_op_mul_1.cc index 730e6c9dd3d..f9de1d8ffb9 100644 --- a/tensorflow/core/kernels/cwise_op_mul_1.cc +++ b/tensorflow/core/kernels/cwise_op_mul_1.cc @@ -31,7 +31,7 @@ REGISTER(BinaryOp, CPU, "Mul", functor::mul, int32); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER4(BinaryOp, GPU, "Mul", functor::mul, Eigen::half, float, double, uint8); #else diff --git a/tensorflow/core/kernels/cwise_op_mul_2.cc b/tensorflow/core/kernels/cwise_op_mul_2.cc index bc3a8407a8d..198a4ac728f 100644 --- a/tensorflow/core/kernels/cwise_op_mul_2.cc +++ b/tensorflow/core/kernels/cwise_op_mul_2.cc @@ -26,7 +26,7 @@ REGISTER8(BinaryOp, CPU, "Mul", functor::mul, int8, uint16, uint32, uint64, int16, int64, complex64, complex128); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER8(BinaryOp, GPU, "Mul", functor::mul, int8, uint16, uint32, uint64, int16, int64, complex64, complex128); #else diff --git a/tensorflow/core/kernels/cwise_op_not_equal_to_1.cc b/tensorflow/core/kernels/cwise_op_not_equal_to_1.cc index 22f30ebeccd..948fbd14553 100644 --- a/tensorflow/core/kernels/cwise_op_not_equal_to_1.cc +++ b/tensorflow/core/kernels/cwise_op_not_equal_to_1.cc @@ -22,7 +22,7 @@ REGISTER7(BinaryOp, CPU, "NotEqual", functor::not_equal_to, uint16, uint32, uint64, qint8, qint16, quint8, quint16); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER4(BinaryOp, GPU, "NotEqual", functor::not_equal_to, float, Eigen::half, double, uint8); #else diff --git a/tensorflow/core/kernels/cwise_op_not_equal_to_2.cc b/tensorflow/core/kernels/cwise_op_not_equal_to_2.cc index dc0adc9c711..cd2a9aed1c4 100644 --- a/tensorflow/core/kernels/cwise_op_not_equal_to_2.cc +++ b/tensorflow/core/kernels/cwise_op_not_equal_to_2.cc @@ -26,7 +26,7 @@ REGISTER6(BinaryOp, CPU, "NotEqual", functor::not_equal_to, int32, int64, complex64, complex128, tstring, bool); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER6(BinaryOp, GPU, "NotEqual", functor::not_equal_to, int8, int16, int64, complex64, complex128, bool); #else diff --git a/tensorflow/core/kernels/cwise_op_pow.cc b/tensorflow/core/kernels/cwise_op_pow.cc index 5132cb87b6a..34276ea4475 100644 --- a/tensorflow/core/kernels/cwise_op_pow.cc +++ b/tensorflow/core/kernels/cwise_op_pow.cc @@ -22,7 +22,7 @@ REGISTER4(BinaryOp, CPU, "Pow", functor::safe_pow, int8, int16, int32, int64); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER3(BinaryOp, GPU, "Pow", functor::pow, float, Eigen::half, double); REGISTER(BinaryOp, GPU, "Pow", functor::safe_pow_ignore_error, int64); #endif diff --git a/tensorflow/core/kernels/cwise_op_right_shift.cc b/tensorflow/core/kernels/cwise_op_right_shift.cc index b72ec4988d9..74b5dc8c334 100644 --- a/tensorflow/core/kernels/cwise_op_right_shift.cc +++ b/tensorflow/core/kernels/cwise_op_right_shift.cc @@ -22,7 +22,7 @@ REGISTER8(BinaryOp, CPU, "RightShift", functor::right_shift, int8, int16, int32, #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER8(BinaryOp, GPU, "RightShift", functor::right_shift, int8, int16, int32, int64, uint8, uint16, uint32, uint64); #else diff --git a/tensorflow/core/kernels/cwise_op_sign.cc b/tensorflow/core/kernels/cwise_op_sign.cc index 7c1e9fb67eb..d821bfb4f61 100644 --- a/tensorflow/core/kernels/cwise_op_sign.cc +++ b/tensorflow/core/kernels/cwise_op_sign.cc @@ -20,7 +20,7 @@ REGISTER8(UnaryOp, CPU, "Sign", functor::sign, float, double, int32, int64, complex64, Eigen::half, bfloat16, complex128); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER6(UnaryOp, GPU, "Sign", functor::sign, float, Eigen::half, double, int64, complex64, complex128); #else diff --git a/tensorflow/core/kernels/cwise_op_sin.cc b/tensorflow/core/kernels/cwise_op_sin.cc index b137e690221..ace83944f11 100644 --- a/tensorflow/core/kernels/cwise_op_sin.cc +++ b/tensorflow/core/kernels/cwise_op_sin.cc @@ -21,7 +21,7 @@ REGISTER6(UnaryOp, CPU, "Sin", functor::sin, float, Eigen::half, bfloat16, #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER3(UnaryOp, GPU, "Sin", functor::sin, float, Eigen::half, double); #endif #endif diff --git a/tensorflow/core/kernels/cwise_op_sinh.cc b/tensorflow/core/kernels/cwise_op_sinh.cc index c93befb5a5d..12a79369f22 100644 --- a/tensorflow/core/kernels/cwise_op_sinh.cc +++ b/tensorflow/core/kernels/cwise_op_sinh.cc @@ -22,7 +22,7 @@ REGISTER5(UnaryOp, CPU, "Sinh", functor::sinh, float, double, bfloat16, #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER2(UnaryOp, GPU, "Sinh", functor::sinh, float, double); #endif #endif diff --git a/tensorflow/core/kernels/cwise_op_squared_difference.cc b/tensorflow/core/kernels/cwise_op_squared_difference.cc index 998104b0b01..29eccb29bab 100644 --- a/tensorflow/core/kernels/cwise_op_squared_difference.cc +++ b/tensorflow/core/kernels/cwise_op_squared_difference.cc @@ -21,7 +21,7 @@ REGISTER8(BinaryOp, CPU, "SquaredDifference", functor::squared_difference, complex128); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER4(BinaryOp, GPU, "SquaredDifference", functor::squared_difference, float, Eigen::half, double, int64); #endif diff --git a/tensorflow/core/kernels/cwise_op_sub.cc b/tensorflow/core/kernels/cwise_op_sub.cc index 1421ede4906..98a37e338ba 100644 --- a/tensorflow/core/kernels/cwise_op_sub.cc +++ b/tensorflow/core/kernels/cwise_op_sub.cc @@ -31,7 +31,7 @@ REGISTER(BinaryOp, CPU, "Sub", functor::sub, int32); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ - !defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED) + !defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED) REGISTER8(BinaryOp, GPU, "Sub", functor::sub, float, Eigen::half, double, int64, complex64, complex128, uint32, uint64); #else