// Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.//// Licensed under the Apache License, Version 2.0 (the "License");// you may not use this file except in compliance with the License.// You may obtain a copy of the License at//// http://www.apache.org/licenses/LICENSE-2.0//// Unless required by applicable law or agreed to in writing, software// distributed under the License is distributed on an "AS IS" BASIS,// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.// See the License for the specific language governing permissions and// limitations under the License.#pragma once#include <memory>#include <thread> // NOLINT#include "paddle/fluid/memory/allocation/allocator.h"namespacepaddle{namespacememory{namespaceallocation{classLockedAllocator:publicUnmanagedAllocator{public:explicitLockedAllocator(std::unique_ptr<Allocator>&&underlying_allocator);std::unique_ptr<Allocation>Allocate(size_tsize,Attrattr=kDefault)override;voidFree(Allocation*allocation)override;boolIsAllocThreadSafe()constoverride;private:std::unique_ptr<UnmanagedAllocator>underlying_allocator_;std::mutexmtx_;};}// namespace allocation}// namespace memory}// namespace paddle