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

Enable the generated version of cos and cosh kernels.

Browse Source 
PiperOrigin-RevId: 354503126
Change-Id: I1a7b6fdc50e0a74b9ffc95a44827372188cad78d
This commit is contained in:
Stephan Herhut 2021-01-29 03:42:41 -08:00 committed by TensorFlower Gardener
parent 65c24bc28f
commit 78250f869b
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_cos.cc
View File

@ -20,8 +20,7 @@ REGISTER6(UnaryOp, CPU, "Cos", functor::cos, 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)
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED)
REGISTER3(UnaryOp, GPU, "Cos", functor::cos, float, Eigen::half, double);
#endif
#endif

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

@ -20,8 +20,7 @@ REGISTER5(UnaryOp, CPU, "Cosh", functor::cosh, float, double, bfloat16,
complex64, complex128);
#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \
!defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED)
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED)
REGISTER2(UnaryOp, GPU, "Cosh", functor::cosh, float, double);
#endif
#endif

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

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

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

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

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

@ -41,6 +41,8 @@ filegroup(
"gpu_op_abs.cc",
"gpu_op_atan.cc",
"gpu_op_atanh.cc",
"gpu_op_cos.cc",
"gpu_op_cosh.cc",
"gpu_op_tan.cc",
"gpu_op_tanh.cc",
],
@ -58,8 +60,6 @@ filegroup(
"gpu_op_complex.cc",
"gpu_op_complex_abs.cc",
"gpu_op_conj.cc",
"gpu_op_cos.cc",
"gpu_op_cosh.cc",
"gpu_op_erf.cc",
"gpu_op_erfc.cc",
"gpu_op_exp.cc",