From 307a41245fe73ff423cf4c2b78a509c6b38f4fda Mon Sep 17 00:00:00 2001
From: Sounak Gupta <sounak.gupta@intel.com>
Date: Thu, 30 Nov 2023 06:53:51 -0800
Subject: [PATCH] Added Intel Data Movers Library to cachelib dependencies

---
 .gitignore               |  2 ++
 BUILD.md                 |  4 +++-
 contrib/build-package.sh | 15 +++++++++++++++
 3 files changed, 20 insertions(+), 1 deletion(-)

diff --git a/.gitignore b/.gitignore
index 9dc75b30e..31c0b2da5 100644
--- a/.gitignore
+++ b/.gitignore
@@ -13,6 +13,7 @@ build-glog/
 build-googletest/
 build-sparsemap/
 build-wangle/
+build-dml/
 
 # Required source-code dependencies, fetched by ./contrib/ scripts.
 cachelib/external/fmt/
@@ -20,3 +21,4 @@ cachelib/external/gflags/
 cachelib/external/glog/
 cachelib/external/googletest/
 cachelib/external/sparsemap/
+cachelib/external/dml/
diff --git a/BUILD.md b/BUILD.md
index efe912659..11f8d2723 100644
--- a/BUILD.md
+++ b/BUILD.md
@@ -22,6 +22,7 @@ These dependencies further require multiple libraries:
 * [https://github.com/google/googletest.git](googletest) - Google Testing Framework
 * [https://github.com/fmtlib/fmt.git](fmt) - open-source formatting library
 * [https://github.com/Tessil/sparse-map.git](sparse-map) - memory efficient hash map and hash set
+* [https://github.com/intel/DML.git](dml) - provides high-performance data manipulation on Intel CPUs
 * And many more libraries, commonly available as installable packages, e.g:
   `boost`, `libevent`, `lz4`, `snappy`, `zlib`, `ssl`, `libunwind`, `libsodium`
 
@@ -148,7 +149,7 @@ options:
 NAME: the dependency to build supported values are:
   googlelog, googleflags, googletest,
   fmt, sparsemap,
-  folly, fizz, wangle, fbthrift,
+  folly, fizz, wangle, fbthrift, dml,
   cachelib
 ```
 
@@ -261,6 +262,7 @@ Example:
 ./contrib/build-package.sh -B googlelog
 ./contrib/build-package.sh -B googleflags
 ./contrib/build-package.sh -B googletest
+./contrib/build-package.sh -B dml
 ./contrib/build-package.sh -B fmt
 ./contrib/build-package.sh -B sparsemap
 ./contrib/build-package.sh -B folly
diff --git a/contrib/build-package.sh b/contrib/build-package.sh
index 310792ca2..2f2efd1e4 100755
--- a/contrib/build-package.sh
+++ b/contrib/build-package.sh
@@ -154,6 +154,21 @@ case "$1" in
     external_git_clone=yes
     ;;
 
+  dml)
+    NAME=dml
+    REPO=https://github.com/intel/DML.git
+    REPODIR=cachelib/external/$NAME
+    SRCDIR=$REPODIR
+    external_git_clone=yes
+    external_git_tag="v1.1.1"
+    cmake_custom_params="-DCMAKE_INSTALL_PREFIX=/usr/ -DDML_BUILD_TESTS=OFF -DLOG_HW_INIT=ON"
+    if test "$debug_build" ; then
+        cmake_custom_params="$cmake_custom_params -DCMAKE_BUILD_TYPE=Debug"
+    else
+        cmake_custom_params="$cmake_custom_params -DCMAKE_BUILD_TYPE=RelWithDebInfo"
+    fi
+    ;;
+
   fmt)
     NAME=fmt
     REPO=https://github.com/fmtlib/fmt.git