423538a816
Summary: - Make MemoryAllocator and its implementations into a Customizable class. - Added a "DefaultMemoryAllocator" which uses new and delete - Added a "CountedMemoryAllocator" that counts the number of allocs and free - Updated the existing tests to use these new allocators - Changed the memkind allocator test into a generic test that can test the various allocators. - Added tests for creating all of the allocators - Added tests to verify/create the JemallocNodumpAllocator using its options. Pull Request resolved: https://github.com/facebook/rocksdb/pull/8980 Reviewed By: zhichao-cao Differential Revision: D32990403 Pulled By: mrambacher fbshipit-source-id: 6fdfe8218c10dd8dfef34344a08201be1fa95c76
45 lines
1.2 KiB
C++
45 lines
1.2 KiB
C++
// Copyright (c) 2011-present, Facebook, Inc. All rights reserved.
|
|
// Copyright (c) 2019 Intel Corporation
|
|
// This source code is licensed under both the GPLv2 (found in the
|
|
// COPYING file in the root directory) and Apache 2.0 License
|
|
// (found in the LICENSE.Apache file in the root directory).
|
|
|
|
#ifdef MEMKIND
|
|
#include <memkind.h>
|
|
#endif // MEMKIND
|
|
|
|
#include "memory/memkind_kmem_allocator.h"
|
|
|
|
namespace ROCKSDB_NAMESPACE {
|
|
Status MemkindKmemAllocator::PrepareOptions(const ConfigOptions& options) {
|
|
std::string message;
|
|
if (!IsSupported(&message)) {
|
|
return Status::NotSupported(message);
|
|
} else {
|
|
return MemoryAllocator::PrepareOptions(options);
|
|
}
|
|
}
|
|
|
|
#ifdef MEMKIND
|
|
void* MemkindKmemAllocator::Allocate(size_t size) {
|
|
void* p = memkind_malloc(MEMKIND_DAX_KMEM, size);
|
|
if (p == NULL) {
|
|
throw std::bad_alloc();
|
|
}
|
|
return p;
|
|
}
|
|
|
|
void MemkindKmemAllocator::Deallocate(void* p) {
|
|
memkind_free(MEMKIND_DAX_KMEM, p);
|
|
}
|
|
|
|
#ifdef ROCKSDB_MALLOC_USABLE_SIZE
|
|
size_t MemkindKmemAllocator::UsableSize(void* p,
|
|
size_t /*allocation_size*/) const {
|
|
return memkind_malloc_usable_size(MEMKIND_DAX_KMEM, p);
|
|
}
|
|
#endif // ROCKSDB_MALLOC_USABLE_SIZE
|
|
#endif // MEMKIND
|
|
|
|
} // namespace ROCKSDB_NAMESPACE
|