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

Change trace level of memory profile in BFCAllocator.

Browse Source 
PiperOrigin-RevId: 297635148
Change-Id: I37cf6886ad3ca33ab4223a62cb7880240e76269b
This commit is contained in:
A. Unique TensorFlower 2020-02-27 10:46:19 -08:00 committed by TensorFlower Gardener
parent e6c4509b36
commit a47a7f4ad1
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/core/common_runtime/bfc_allocator.cc
View File

@ -443,6 +443,12 @@ void* BFCAllocator::AllocateRawInternal(size_t unused_alignment,
void BFCAllocator::AddTraceMe(absl::string_view traceme_name,
int64 requested_bytes) {
// Internal users will see the memory profile with default trace level.
auto traceme_level = profiler::TraceMeLevel::kVerbose;
#ifdef PLATFORM_GOOGLE
traceme_level = profiler::TraceMeLevel::kInfo;
#endif
tensorflow::profiler::TraceMe trace_me(
[&]() EXCLUSIVE_LOCKS_REQUIRED(lock_) {
AllocatorStats stats = stats_;
@ -459,7 +465,7 @@ void BFCAllocator::AddTraceMe(absl::string_view traceme_name,
#endif
"#");
},
/*level=*/profiler::TraceMeLevel::kInfo);
traceme_level);
}
void* BFCAllocator::FindChunkPtr(BinNum bin_num, size_t rounded_bytes,