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

Enable the generated versions of exp and expm1.

Browse Source 
PiperOrigin-RevId: 358170449
Change-Id: Ia8ca6ce9c1dfecc429df5aa8c015493d633467cd
This commit is contained in:
Stephan Herhut 2021-02-18 07:08:42 -08:00 committed by TensorFlower Gardener
parent 611a68e5bd
commit 8e81901908
5 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -20,8 +20,7 @@ REGISTER6(UnaryOp, CPU, "Exp", functor::exp, float, Eigen::half, bfloat16,
double, complex64, complex128); double, complex64, complex128);
#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_KERNELS_ENABLED)
REGISTER5(UnaryOp, GPU, "Exp", functor::exp, float, Eigen::half, double, REGISTER5(UnaryOp, GPU, "Exp", functor::exp, float, Eigen::half, double,
complex64, complex128); complex64, complex128);
#endif #endif

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

@ -19,8 +19,7 @@ namespace tensorflow {
REGISTER6(UnaryOp, CPU, "Expm1", functor::expm1, float, Eigen::half, bfloat16, REGISTER6(UnaryOp, CPU, "Expm1", functor::expm1, float, Eigen::half, bfloat16,
double, complex64, complex128); double, complex64, complex128);
#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_KERNELS_ENABLED)
REGISTER3(UnaryOp, GPU, "Expm1", functor::expm1, float, Eigen::half, double); REGISTER3(UnaryOp, GPU, "Expm1", functor::expm1, float, Eigen::half, double);
#endif #endif
#endif #endif

2
tensorflow/core/kernels/cwise_op_gpu_exp.cu.cc
View File

@ -19,7 +19,9 @@ limitations under the License.
namespace tensorflow { namespace tensorflow {
namespace functor { namespace functor {
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED)
DEFINE_UNARY5(exp, Eigen::half, float, double, complex64, complex128); DEFINE_UNARY5(exp, Eigen::half, float, double, complex64, complex128);
#endif
} // namespace functor } // namespace functor
} // namespace tensorflow } // namespace tensorflow

2
tensorflow/core/kernels/cwise_op_gpu_expm1.cu.cc
View File

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

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

@ -56,6 +56,8 @@ filegroup(
"gpu_op_digamma.cc", "gpu_op_digamma.cc",
"gpu_op_erf.cc", "gpu_op_erf.cc",
"gpu_op_erfc.cc", "gpu_op_erfc.cc",
"gpu_op_exp.cc",
"gpu_op_expm1.cc",
"gpu_op_floor.cc", "gpu_op_floor.cc",
"gpu_op_imag.cc", "gpu_op_imag.cc",
"gpu_op_invert.cc", "gpu_op_invert.cc",
@ -82,8 +84,6 @@ filegroup(
filegroup( filegroup(
name = "experimental_unary_gpu_kernel_srcs", name = "experimental_unary_gpu_kernel_srcs",
srcs = [ srcs = [
"gpu_op_exp.cc",
"gpu_op_expm1.cc",
"gpu_op_sign.cc", "gpu_op_sign.cc",
], ],
compatible_with = get_compatible_with_cloud(), compatible_with = get_compatible_with_cloud(),