import quincy beta 17.1.0

[ceph.git] / ceph / src / rocksdb / memory / memkind_kmem_allocator.h

diff --git a/ceph/src/rocksdb/memory/memkind_kmem_allocator.h b/ceph/src/rocksdb/memory/memkind_kmem_allocator.h
new file mode 100644 (file)
index 0000000..8bf3c6c
--- /dev/null
+++ b/ceph/src/rocksdb/memory/memkind_kmem_allocator.h
@@ -0,0 +1,27 @@
+//  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).
+
+#pragma once
+
+#ifdef MEMKIND
+
+#include <memkind.h>
+#include "rocksdb/memory_allocator.h"
+
+namespace rocksdb {
+
+class MemkindKmemAllocator : public MemoryAllocator {
+ public:
+  const char* Name() const override { return "MemkindKmemAllocator"; };
+  void* Allocate(size_t size) override;
+  void Deallocate(void* p) override;
+#ifdef ROCKSDB_MALLOC_USABLE_SIZE
+  size_t UsableSize(void* p, size_t /*allocation_size*/) const override;
+#endif
+};
+
+}  // namespace rocksdb
+#endif  // MEMKIND