add tbbmalloc definition

2 files changed, 26 insertions, 1 deletions

diff --git a/src/allocators/all.py b/src/allocators/all.py
index ccf48b8..9afacf7 100644
--- a/src/allocators/all.py
+++ b/src/allocators/all.py
@@ -9,7 +9,8 @@ from src.allocators.mesh import mesh
 from src.allocators.scalloc import scalloc
 from src.allocators.supermalloc import supermalloc
 from src.allocators.llalloc import llalloc
+from src.allocators.tbbmalloc import tbbmalloc
 
 
 allocators = [*src.allocators.glibcs.allocators, tcmalloc, tcmalloc_nofs,
-              jemalloc, hoard, mesh, supermalloc, scalloc, llalloc]
+              jemalloc, hoard, mesh, supermalloc, scalloc, llalloc, tbbmalloc]
diff --git a/src/allocators/tbbmalloc.py b/src/allocators/tbbmalloc.py
new file mode 100644
index 0000000..420803b
--- /dev/null
+++ b/src/allocators/tbbmalloc.py
@@ -0,0 +1,24 @@
+import src.allocator
+
+version = "2019_U8"
+
+source = src.allocator.Allocator_Sources("tbb",
+                         ["git clone https://github.com/intel/tbb.git"],
+                         ["git checkout {}".format(version)],
+                         ["git stash"])
+
+
+class TBBMalloc (src.allocator.Allocator):
+    """TCMalloc definition for allocbench"""
+    def __init__(self, name, **kwargs):
+
+        kwargs["sources"] = source
+        kwargs["LD_PRELOAD"] = "{dir}/libtbbmalloc.so"
+        kwargs["build_cmds"] = ["cd {srcdir}; make tbbmalloc -j4",
+                                "mkdir -p {dir}",
+                                'ln -f -s $(find {srcdir} -executable -name "*libtbbmalloc.so*") {dir}/libtbbmalloc.so']
+
+        super().__init__(name, **kwargs)
+
+
+tbbmalloc = TBBMalloc("tbbmalloc")