From f5598990263b02682a643d3178725d4c10e54ba8 Mon Sep 17 00:00:00 2001 From: locke Date: Sun, 13 Nov 2022 13:10:25 +0800 Subject: [PATCH] add missing lock_ to GPUAdjustableAllocator --- tensorflow/core/common_runtime/gpu/gpu_adjustable_allocator.h | 1 + 1 file changed, 1 insertion(+) diff --git a/tensorflow/core/common_runtime/gpu/gpu_adjustable_allocator.h b/tensorflow/core/common_runtime/gpu/gpu_adjustable_allocator.h index 4d675aad5ec..9bca4cf9d98 100644 --- a/tensorflow/core/common_runtime/gpu/gpu_adjustable_allocator.h +++ b/tensorflow/core/common_runtime/gpu/gpu_adjustable_allocator.h @@ -46,6 +46,7 @@ class GPUAdjustableAllocator final { size_t FreeEmptyMemory(size_t target_memory_bytes, BFCAllocator* bfc_allocator) EXCLUSIVE_LOCKS_REQUIRED(lock_); + mutex lock_; }; } // namespace tensorflow