aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/common_runtime
diff options
context:
space:
mode:
authorGravatar Vijay Vasudevan <vrv@google.com>2016-11-03 17:07:01 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-11-03 18:24:53 -0700
commit818993c7751601527d662d2417f220e4e856e4ef (patch)
treea9cb33d6332f3e37d740cd6eb6984a1837714237 /tensorflow/core/common_runtime
parenta19c425536bba29997807bbbd5ed43386d3cb7bd (diff)
Merge changes from github.
Change: 138143557
Diffstat (limited to 'tensorflow/core/common_runtime')
-rw-r--r--tensorflow/core/common_runtime/gpu/pool_allocator.h2
-rw-r--r--tensorflow/core/common_runtime/simple_placer.cc6
2 files changed, 5 insertions, 3 deletions
diff --git a/tensorflow/core/common_runtime/gpu/pool_allocator.h b/tensorflow/core/common_runtime/gpu/pool_allocator.h
index b2f0265145..5842758f0e 100644
--- a/tensorflow/core/common_runtime/gpu/pool_allocator.h
+++ b/tensorflow/core/common_runtime/gpu/pool_allocator.h
@@ -173,7 +173,7 @@ class BasicCPUAllocator : public SubAllocator {
void* Alloc(size_t alignment, size_t num_bytes) override {
return port::aligned_malloc(num_bytes, alignment);
}
- void Free(void* ptr, size_t num_bytes) override { free(ptr); }
+ void Free(void* ptr, size_t num_bytes) override { port::aligned_free(ptr); }
};
// Allocator for pinned CPU RAM that is made known to CUDA for the
diff --git a/tensorflow/core/common_runtime/simple_placer.cc b/tensorflow/core/common_runtime/simple_placer.cc
index fda429b52a..721e73898b 100644
--- a/tensorflow/core/common_runtime/simple_placer.cc
+++ b/tensorflow/core/common_runtime/simple_placer.cc
@@ -815,9 +815,11 @@ void SimplePlacer::AssignAndLog(const string& assigned_device,
node->set_assigned_device_name(assigned_device);
// Log placement if log_device_placement is set.
if (options_ && options_->config.log_device_placement()) {
- printf("%s: %s\n", node->name().c_str(),
+ printf("%s: (%s): %s\n", node->name().c_str(),
+ node->type_string().c_str(),
node->assigned_device_name().c_str());
- LOG(INFO) << node->name() << ": " << node->assigned_device_name();
+ LOG(INFO) << node->name() << ": " << "(" << node->type_string() << ")"
+ << node->assigned_device_name();
}
}