diff options
author | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-27 13:04:31 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-27 13:04:31 -0700 |
commit | c898e63d07fc63315be98f0772736e5d7f2fb44c (patch) | |
tree | f250fcb1b35714d92afb371bf322f978ddcdfa6e /tensorflow/core/common_runtime | |
parent | 1084594657a5d139102ac794f84d1427a710e39a (diff) | |
parent | 937ad7c27f0d289067c935543d282e5ac5a310b1 (diff) |
Merge pull request #22286 from Intel-tensorflow:nhasabni/unit-test-fixes
PiperOrigin-RevId: 214821528
Diffstat (limited to 'tensorflow/core/common_runtime')
-rw-r--r-- | tensorflow/core/common_runtime/mkl_cpu_allocator.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/common_runtime/mkl_cpu_allocator.h b/tensorflow/core/common_runtime/mkl_cpu_allocator.h index 538a70668a..429b19599b 100644 --- a/tensorflow/core/common_runtime/mkl_cpu_allocator.h +++ b/tensorflow/core/common_runtime/mkl_cpu_allocator.h @@ -251,6 +251,7 @@ class MklCPUAllocator : public Allocator { // max_alloc_size from large_size_allocator would be the maximum // size allocated by MklCPUAllocator. stats->max_alloc_size = l_stats.max_alloc_size; + stats->bytes_limit = std::max(s_stats.bytes_limit, l_stats.bytes_limit); } void ClearStats() override { |