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:
parent
8db82bf5c4
commit
5735726ed6
@ -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",
|
||||
|
@ -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 {
|
||||
|
Loading…
Reference in New Issue
Block a user