Add back the "no_rocm" tag to cwise_unary_op target.
It seems in the Kokoro build environment, hexdump is not available. PiperOrigin-RevId: 333042998 Change-Id: I2a5fe58ffed008f37ddb807378cc6e4f0d03e33f
This commit is contained in:
parent
a0a4b95f50
commit
712fd5cfe4
@ -41,6 +41,9 @@ tf_kernel_library(
|
|||||||
"cwise_op_gpu_base.h",
|
"cwise_op_gpu_base.h",
|
||||||
"cwise_op_gpu_tanh.cc",
|
"cwise_op_gpu_tanh.cc",
|
||||||
]),
|
]),
|
||||||
|
# Building abs_kernels and tanh_kernels requires hexdump on the ROCM
|
||||||
|
# platform, which might not be available.
|
||||||
|
tags = ["no_rocm"],
|
||||||
deps = if_cuda_or_rocm([
|
deps = if_cuda_or_rocm([
|
||||||
":abs_kernels",
|
":abs_kernels",
|
||||||
":tanh_kernels",
|
":tanh_kernels",
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user