From 6fb26167f0eb92bc6fe8880e9fee23b2301e79ad Mon Sep 17 00:00:00 2001
From: Jefferson Fialho <jfialho@ibm.com>
Date: Thu, 12 Dec 2024 09:31:40 -0300
Subject: [PATCH] install numactl to enable fastsafetensors

Signed-off-by: Jefferson Fialho <jfialho@ibm.com>
---
 Dockerfile.ubi | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/Dockerfile.ubi b/Dockerfile.ubi
index 8bb80a299ccc4..02e3f3ff5b157 100644
--- a/Dockerfile.ubi
+++ b/Dockerfile.ubi
@@ -43,9 +43,16 @@ RUN curl -Lo /etc/yum.repos.d/cuda-rhel9.repo \
         https://developer.download.nvidia.com/compute/cuda/repos/rhel9/x86_64/cuda-rhel9.repo
 
 RUN microdnf install -y \
-        cuda-nvcc-12-4 cuda-nvtx-12-4 cuda-libraries-devel-12-4 && \
+        cuda-nvcc-12-4 cuda-nvtx-12-4 cuda-libraries-devel-12-4 \
+        numactl-libs && \
+        rpm -ivh https://yum.oracle.com/repo/OracleLinux/OL9/appstream/x86_64/getPackage/numactl-devel-2.0.18-2.el9.x86_64.rpm && \
     microdnf clean all
 
+# Search for numa.h in common locations
+RUN echo "### Searching for numa.h in common directories ###" && \
+    find /usr /usr/local /opt /lib -name numa.h | tee /dev/stderr && \
+    echo "### END OF NUMA.H SEARCH ###"
+
 ENV CUDA_HOME="/usr/local/cuda" \
     PATH="${CUDA_HOME}/bin:${PATH}" \
     LD_LIBRARY_PATH="${CUDA_HOME}/lib64:${CUDA_HOME}/extras/CUPTI/lib64:${LD_LIBRARY_PATH}"