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

Enable the generated kernel for angle.

Browse Source 
PiperOrigin-RevId: 355645510
Change-Id: I5e8a91615186098f596269bed11c151c70eb5020
This commit is contained in:
Stephan Herhut 2021-02-04 09:37:14 -08:00 committed by TensorFlower Gardener
parent 868276741a
commit e743fcee33
3 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -27,8 +27,7 @@ REGISTER_COMPLEX(CPU, float, complex64);
REGISTER_COMPLEX(CPU, double, complex128);
#if GOOGLE_CUDA
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \
!defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED)
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED)
REGISTER_COMPLEX(GPU, float, complex64);
REGISTER_COMPLEX(GPU, double, complex128);
#endif

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

@ -20,8 +20,7 @@ limitations under the License.
namespace tensorflow {
namespace functor {
#if GOOGLE_CUDA
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \
!defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED)
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED)
DEFINE_UNARY2(get_angle, complex64, complex128);
#endif
#endif

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

@ -41,6 +41,7 @@ filegroup(
"gpu_op_abs.cc",
"gpu_op_acos.cc",
"gpu_op_acosh.cc",
"gpu_op_angle.cc",
"gpu_op_asin.cc",
"gpu_op_asinh.cc",
"gpu_op_atan.cc",
@ -78,7 +79,6 @@ filegroup(
filegroup(
name = "experimental_unary_kernel_srcs",
srcs = [
"gpu_op_angle.cc",
"gpu_op_digamma.cc",
"gpu_op_exp.cc",
"gpu_op_expm1.cc",