Disable unranked kernels from tests.
PiperOrigin-RevId: 342588059 Change-Id: Icf48544ffc325e1b353389072f9f1d634d00de77
This commit is contained in:
parent
34ef6a6805
commit
6468a1d17a
@ -82,7 +82,11 @@ cc_library(
|
||||
tf_kernel_library(
|
||||
name = "cwise_unary_op",
|
||||
srcs = [":unary_kernel_srcs"],
|
||||
tags = ["manual"],
|
||||
# TODO(herhut): Remove no_rocm once kernels build again.
|
||||
tags = [
|
||||
"manual",
|
||||
"no_rocm",
|
||||
],
|
||||
deps = if_mlir_unranked_kernels_enabled(
|
||||
if_false = [
|
||||
":abs_kernels",
|
||||
@ -115,7 +119,11 @@ tf_kernel_library(
|
||||
tf_kernel_library(
|
||||
name = "cwise_binary_op",
|
||||
srcs = ["unranked_gpu_add.cc"],
|
||||
tags = ["manual"],
|
||||
# TODO(herhut): Remove no_rocm once kernels build again.
|
||||
tags = [
|
||||
"manual",
|
||||
"no_rocm",
|
||||
],
|
||||
deps = [
|
||||
":addv2_unranked_kernels",
|
||||
":unranked_op_gpu_base",
|
||||
@ -126,7 +134,11 @@ tf_kernel_library(
|
||||
tf_kernel_library(
|
||||
name = "cwise_op",
|
||||
srcs = [],
|
||||
tags = ["no_rocm"],
|
||||
# TODO(herhut): Remove manual once kernels build again.
|
||||
tags = [
|
||||
"manual",
|
||||
"no_rocm",
|
||||
],
|
||||
# Technically these libraries don't need --config=cuda or --config=rocm,
|
||||
# but we want to avoid building them if they are not needed.
|
||||
deps = if_cuda_or_rocm([
|
||||
|
Loading…
Reference in New Issue
Block a user