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

Enable the generated kernel for log and log1p.

Browse Source 
PiperOrigin-RevId: 355013308
Change-Id: If49e963f505c3db3ad323fd2876772c673c8108f
This commit is contained in:
Stephan Herhut 2021-02-01 13:28:24 -08:00 committed by TensorFlower Gardener
parent 53ab21aa86
commit 732f1e2c94
5 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/kernels/cwise_op_gpu_log.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(log, Eigen::half, float, double); DEFINE_UNARY3(log, Eigen::half, float, double);
#endif
} // namespace functor } // namespace functor
} // namespace tensorflow } // namespace tensorflow

2
tensorflow/core/kernels/cwise_op_gpu_log1p.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(log1p, Eigen::half, float, double); DEFINE_UNARY3(log1p, Eigen::half, float, double);
#endif
} // namespace functor } // namespace functor
} // namespace tensorflow } // namespace tensorflow

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

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

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

@ -20,8 +20,7 @@ REGISTER6(UnaryOp, CPU, "Log1p", functor::log1p, 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_GPU_KERNELS_ENABLED)
REGISTER3(UnaryOp, GPU, "Log1p", functor::log1p, float, Eigen::half, double); REGISTER3(UnaryOp, GPU, "Log1p", functor::log1p, float, Eigen::half, double);
#endif #endif
#endif #endif

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

@ -52,6 +52,8 @@ 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_log.cc",
"gpu_op_log1p.cc",
"gpu_op_logical_not.cc", "gpu_op_logical_not.cc",
"gpu_op_neg.cc", "gpu_op_neg.cc",
"gpu_op_real.cc", "gpu_op_real.cc",
@ -76,8 +78,6 @@ filegroup(
"gpu_op_is_inf.cc", "gpu_op_is_inf.cc",
"gpu_op_is_nan.cc", "gpu_op_is_nan.cc",
"gpu_op_lgamma.cc", "gpu_op_lgamma.cc",
"gpu_op_log.cc",
"gpu_op_log1p.cc",
"gpu_op_rsqrt.cc", "gpu_op_rsqrt.cc",
"gpu_op_sign.cc", "gpu_op_sign.cc",
"gpu_op_sin.cc", "gpu_op_sin.cc",