From 4b700752f301402eddee1114c95044df74968855 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Tue, 21 Apr 2020 19:27:49 -0700 Subject: [PATCH] Using "is_cuda_is_configured_compat" to select the cuda libraries. PiperOrigin-RevId: 307728709 Change-Id: I90edb98ca5a2eb641f27e3fa54d6a6cbcaf8420e --- tensorflow/core/profiler/internal/gpu/BUILD | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tensorflow/core/profiler/internal/gpu/BUILD b/tensorflow/core/profiler/internal/gpu/BUILD index 24568091e88..c6fe4d77031 100644 --- a/tensorflow/core/profiler/internal/gpu/BUILD +++ b/tensorflow/core/profiler/internal/gpu/BUILD @@ -91,8 +91,9 @@ tf_cuda_library( deps = [ "//tensorflow/core:lib", "//tensorflow/core:platform_base", + ] + if_cuda_is_configured_compat([ "//tensorflow/stream_executor/cuda:cupti_stub", - ], + ]), ) tf_cuda_library( @@ -103,8 +104,9 @@ tf_cuda_library( visibility = ["//visibility:public"], deps = [ ":cupti_interface", + ] + if_cuda_is_configured_compat([ "//tensorflow/stream_executor/cuda:cupti_stub", - ], + ]), ) tf_cuda_library(