Fix up the gpu_bfc_allocator target:

- Avoid duplicate symbols by establishing proper dependency
 - Avoid error about gpu_id.h header by dropping the include

PiperOrigin-RevId: 249849197
This commit is contained in:
George Karpenkov 2019-05-24 09:51:18 -07:00 committed by TensorFlower Gardener
parent 8db82bf5c4
commit 5735726ed6
2 changed files with 1 additions and 2 deletions

View File

@ -3494,7 +3494,6 @@ GPU_RUNTIME_HEADERS = [
tf_cuda_library(
name = "gpu_runtime_impl",
srcs = [
"common_runtime/gpu/gpu_bfc_allocator.cc",
"common_runtime/gpu/gpu_cudamalloc_allocator.cc",
"common_runtime/gpu/gpu_debug_allocator.cc",
"common_runtime/gpu/gpu_device.cc",
@ -3512,6 +3511,7 @@ tf_cuda_library(
":core_cpu_lib",
":framework",
":framework_internal",
":gpu_bfc_allocator",
":gpu_id_impl",
":gpu_init_impl",
":gpu_lib",

View File

@ -15,7 +15,6 @@ limitations under the License.
#include "tensorflow/core/common_runtime/gpu/gpu_bfc_allocator.h"
#include "tensorflow/core/common_runtime/gpu/gpu_id.h"
#include "tensorflow/core/lib/strings/strcat.h"
namespace tensorflow {