diff options
author | 2016-11-03 17:07:01 -0800 | |
---|---|---|
committer | 2016-11-03 18:24:53 -0700 | |
commit | 818993c7751601527d662d2417f220e4e856e4ef (patch) | |
tree | a9cb33d6332f3e37d740cd6eb6984a1837714237 /tensorflow/core/common_runtime | |
parent | a19c425536bba29997807bbbd5ed43386d3cb7bd (diff) |
Merge changes from github.
Change: 138143557
Diffstat (limited to 'tensorflow/core/common_runtime')
-rw-r--r-- | tensorflow/core/common_runtime/gpu/pool_allocator.h | 2 | ||||
-rw-r--r-- | tensorflow/core/common_runtime/simple_placer.cc | 6 |
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(); } } |