aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/common_runtime
diff options
context:
space:
mode:
authorGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-27 13:04:31 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-27 13:04:31 -0700
commitc898e63d07fc63315be98f0772736e5d7f2fb44c (patch)
treef250fcb1b35714d92afb371bf322f978ddcdfa6e /tensorflow/core/common_runtime
parent1084594657a5d139102ac794f84d1427a710e39a (diff)
parent937ad7c27f0d289067c935543d282e5ac5a310b1 (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.h1
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 {