experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Move GPU-specific dependencies of core/grappler:devices into

Browse Source 
cuda_deps.

Fix #includes and deps of contrib/verbs:verbs_util, in particular
removing an unnecessary #include of gpu_util.h that relied on a
transitive dependency through :devices.

PiperOrigin-RevId: 169732234
This commit is contained in:
A. Unique TensorFlower 2017-09-22 13:48:33 -07:00 committed by TensorFlower Gardener
parent 7766611a19
commit 0d86463016
4 changed files with 11 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/contrib/verbs/BUILD
View File

@ -50,11 +50,8 @@ cc_library(
srcs = ["verbs_util.cc"],
hdrs = ["verbs_util.h"],
deps = [
"//tensorflow/core:core_cpu_internal",
"//tensorflow/core:framework",
"//tensorflow/core:gpu_runtime",
"//tensorflow/core:lib",
"//tensorflow/core:lib_internal",
],
)

8
tensorflow/contrib/verbs/verbs_util.cc
View File

@ -15,9 +15,13 @@ limitations under the License.
#include "tensorflow/contrib/verbs/verbs_util.h"
#include "tensorflow/core/common_runtime/gpu/gpu_util.h"
#include "tensorflow/core/lib/core/notification.h"
#include <vector>
#include "tensorflow/core/lib/core/stringpiece.h"
#include "tensorflow/core/lib/strings/numbers.h"
#include "tensorflow/core/lib/strings/str_util.h"
#include "tensorflow/core/lib/strings/strcat.h"
namespace tensorflow {
// static

6
tensorflow/contrib/verbs/verbs_util.h
View File

@ -18,14 +18,10 @@ limitations under the License.
#include <string>
#include "tensorflow/core/common_runtime/device.h"
#include "tensorflow/core/framework/tensor.h"
#include "tensorflow/core/lib/core/status.h"
#include "tensorflow/core/framework/types.h"
namespace tensorflow {
class TensorProto;
class VerbsUtil {
public:
static string AppendStepidToKey(const string& key, int64 step_id);

6
tensorflow/core/grappler/BUILD
View File

@ -55,12 +55,14 @@ tf_cuda_library(
name = "devices",
srcs = ["devices.cc"],
hdrs = ["devices.h"],
cuda_deps = [
"//tensorflow/core:gpu_init",
"//tensorflow/core:stream_executor",
],
visibility = ["//visibility:public"],
deps = [
"//tensorflow/core:gpu_init",
"//tensorflow/core:lib",
"//tensorflow/core:lib_internal",
"//tensorflow/core:stream_executor",
],
)