experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Enable the generated version of maximum and minimum.

Browse Source 
PiperOrigin-RevId: 361091405
Change-Id: I6bcdf971aec2ec8788c92ba55f14b475eda5a2a8
This commit is contained in:
Stephan Herhut 2021-03-05 00:45:03 -08:00 committed by TensorFlower Gardener
parent e6966f066a
commit 548976c0de
5 changed files with 6 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/core/kernels/cwise_op_gpu_maximum.cu.cc
View File

@ -19,8 +19,7 @@ limitations under the License.
namespace tensorflow {
namespace functor {
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \
!defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED)
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED)
DEFINE_BINARY6(maximum, Eigen::half, float, double, uint8, int16, int64);
#else
DEFINE_BINARY1(maximum, uint8);

3
tensorflow/core/kernels/cwise_op_gpu_minimum.cu.cc
View File

@ -19,8 +19,7 @@ limitations under the License.
namespace tensorflow {
namespace functor {
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \
!defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED)
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED)
DEFINE_BINARY6(minimum, Eigen::half, float, double, uint8, int16, int64);
#else
DEFINE_BINARY1(minimum, uint8);

3
tensorflow/core/kernels/cwise_op_maximum.cc
View File

@ -19,8 +19,7 @@ namespace tensorflow {
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_KERNELS_ENABLED)
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED)
REGISTER6(BinaryOp, GPU, "Maximum", functor::maximum, float, Eigen::half,
double, uint8, int16, int64);
#else

3
tensorflow/core/kernels/cwise_op_minimum.cc
View File

@ -19,8 +19,7 @@ namespace tensorflow {
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_KERNELS_ENABLED)
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED)
REGISTER6(BinaryOp, GPU, "Minimum", functor::minimum, float, Eigen::half,
double, uint8, int16, int64);
#else

4
tensorflow/core/kernels/mlir_generated/BUILD
View File

@ -109,6 +109,8 @@ filegroup(
"gpu_op_complex.cc",
"gpu_op_logical_and.cc",
"gpu_op_logical_or.cc",
"gpu_op_maximum.cc",
"gpu_op_minimum.cc",
],
compatible_with = get_compatible_with_cloud(),
)
@ -125,8 +127,6 @@ filegroup(
"gpu_op_left_shift.cc",
"gpu_op_less.cc",
"gpu_op_less_equal.cc",
"gpu_op_maximum.cc",
"gpu_op_minimum.cc",
"gpu_op_mul.cc",
"gpu_op_not_equal.cc",
"gpu_op_pow.cc",