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

Enable generated versions of is_inf, is_nan and is_finite.

Browse Source 
PiperOrigin-RevId: 355115941
Change-Id: I3fa22fac95945e651101dfb511abd0d2ba7b781e
This commit is contained in:
Stephan Herhut 2021-02-02 01:10:10 -08:00 committed by TensorFlower Gardener
parent 802b20fdb8
commit 23b1029868
7 changed files with 9 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -19,8 +19,7 @@ limitations under the License.
namespace tensorflow { namespace tensorflow {
namespace functor { namespace functor {
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED)
!defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED)
DEFINE_UNARY3(isfinite, Eigen::half, float, double); DEFINE_UNARY3(isfinite, Eigen::half, float, double);
#endif #endif
} // namespace functor } // namespace functor

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

@ -19,8 +19,7 @@ limitations under the License.
namespace tensorflow { namespace tensorflow {
namespace functor { namespace functor {
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED)
!defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED)
DEFINE_UNARY3(isinf, Eigen::half, float, double); DEFINE_UNARY3(isinf, Eigen::half, float, double);
#endif #endif
} // namespace functor } // namespace functor

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

@ -19,8 +19,7 @@ limitations under the License.
namespace tensorflow { namespace tensorflow {
namespace functor { namespace functor {
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED)
!defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED)
DEFINE_UNARY3(isnan, Eigen::half, float, double); DEFINE_UNARY3(isnan, Eigen::half, float, double);
#endif #endif
} // namespace functor } // namespace functor

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

@ -20,8 +20,7 @@ REGISTER4(UnaryOp, CPU, "IsFinite", functor::isfinite, float, Eigen::half,
bfloat16, double); bfloat16, double);
#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED)
!defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED)
REGISTER3(UnaryOp, GPU, "IsFinite", functor::isfinite, float, Eigen::half, REGISTER3(UnaryOp, GPU, "IsFinite", functor::isfinite, float, Eigen::half,
double); double);
#endif #endif

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

@ -20,8 +20,7 @@ REGISTER4(UnaryOp, CPU, "IsInf", functor::isinf, float, Eigen::half, bfloat16,
double); double);
#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED)
!defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED)
REGISTER3(UnaryOp, GPU, "IsInf", functor::isinf, float, Eigen::half, double); REGISTER3(UnaryOp, GPU, "IsInf", functor::isinf, float, Eigen::half, double);
#endif #endif
#endif #endif

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

@ -20,8 +20,7 @@ REGISTER4(UnaryOp, CPU, "IsNan", functor::isnan, float, Eigen::half, double,
bfloat16); bfloat16);
#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED)
!defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED)
REGISTER3(UnaryOp, GPU, "IsNan", functor::isnan, float, Eigen::half, double); REGISTER3(UnaryOp, GPU, "IsNan", functor::isnan, float, Eigen::half, double);
#endif #endif
#endif #endif

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

@ -52,6 +52,9 @@ filegroup(
"gpu_op_floor.cc", "gpu_op_floor.cc",
"gpu_op_imag.cc", "gpu_op_imag.cc",
"gpu_op_invert.cc", "gpu_op_invert.cc",
"gpu_op_is_finite.cc",
"gpu_op_is_inf.cc",
"gpu_op_is_nan.cc",
"gpu_op_log.cc", "gpu_op_log.cc",
"gpu_op_log1p.cc", "gpu_op_log1p.cc",
"gpu_op_logical_not.cc", "gpu_op_logical_not.cc",
@ -76,9 +79,6 @@ filegroup(
"gpu_op_asinh.cc", "gpu_op_asinh.cc",
"gpu_op_exp.cc", "gpu_op_exp.cc",
"gpu_op_expm1.cc", "gpu_op_expm1.cc",
"gpu_op_is_finite.cc",
"gpu_op_is_inf.cc",
"gpu_op_is_nan.cc",
"gpu_op_lgamma.cc", "gpu_op_lgamma.cc",
"gpu_op_sign.cc", "gpu_op_sign.cc",
"gpu_op_sin.cc", "gpu_op_sin.cc",