Skip to content

Commit

Permalink
changing if_cuda back to if_cuda_is_configured.
Browse files Browse the repository at this point in the history
This change is needed to fix the broken `--config=rocm` build (post PR 26722 merge)
  • Loading branch information
deven-amd committed Apr 24, 2019
1 parent 70800b7 commit caf27f4
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions tensorflow/core/kernels/BUILD
Expand Up @@ -4029,7 +4029,7 @@ tf_kernel_library(
tf_kernel_library(
name = "bias_op",
prefix = "bias_op",
deps = NN_DEPS + [":redux_functor"] + if_cuda([
deps = NN_DEPS + [":redux_functor"] + if_cuda_is_configured([
":reduction_ops",
"@cub_archive//:cub",
"//tensorflow/core:stream_executor",
Expand Down Expand Up @@ -4069,7 +4069,7 @@ tf_kernel_library(
tf_kernel_library(
name = "softmax_op",
prefix = "softmax_op",
deps = NN_DEPS + if_cuda([
deps = NN_DEPS + if_cuda_is_configured([
":reduction_ops",
"@cub_archive//:cub",
]) + if_rocm_is_configured([
Expand Down Expand Up @@ -4738,7 +4738,7 @@ tf_kernel_library(
deps = SPARSE_DEPS + [
":bounds_check",
"//third_party/eigen3",
] + if_cuda([
] + if_cuda_is_configured([
":reduction_ops",
"@cub_archive//:cub",
]) + if_rocm_is_configured([
Expand Down Expand Up @@ -5235,7 +5235,7 @@ tf_kernel_library(
"//tensorflow/core:framework",
"//tensorflow/core:lib",
"//tensorflow/core:lib_internal",
] + if_cuda([
] + if_cuda_is_configured([
":reduction_ops",
"@cub_archive//:cub",
]) + if_rocm_is_configured([
Expand Down

0 comments on commit caf27f4

Please sign in to comment.