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

Don't depend on MLIR generated kernels on ROCM platform.

Browse Source 
They don't work yet in the Kokoro environment because of missing the hexdump
tool.

PiperOrigin-RevId: 333054051
Change-Id: Ia920d759d216d23a565ee36cd58f54eb805be3b5
This commit is contained in:
Adrian Kuegel 2020-09-22 04:53:28 -07:00 committed by TensorFlower Gardener
parent 712fd5cfe4
commit d7b9b383e2
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/compiler/mlir/tools/kernel_gen/BUILD
View File

@ -11,10 +11,6 @@ load(
"@local_config_rocm//rocm:build_defs.bzl", "@local_config_rocm//rocm:build_defs.bzl",
"if_rocm_is_configured", "if_rocm_is_configured",
) )
load(
"//tensorflow/stream_executor:build_defs.bzl",
"if_gpu_is_configured",
)
package( package(
default_visibility = [":friends"], default_visibility = [":friends"],
@ -123,7 +119,7 @@ tf_cc_test(
deps = [ deps = [
"//tensorflow/core:test", "//tensorflow/core:test",
"//tensorflow/core:test_main", "//tensorflow/core:test_main",
] + if_gpu_is_configured([ ] + if_cuda_is_configured([
"//tensorflow/core/kernels/mlir_generated:abs_kernels", "//tensorflow/core/kernels/mlir_generated:abs_kernels",
"//tensorflow/core/kernels/mlir_generated:ceil_kernels", "//tensorflow/core/kernels/mlir_generated:ceil_kernels",
"//tensorflow/core/kernels/mlir_generated:cos_kernels", "//tensorflow/core/kernels/mlir_generated:cos_kernels",