From 98adad4154a895564988928190094568c43e8b97 Mon Sep 17 00:00:00 2001
From: Theodore Chang <tlcfem@gmail.com>
Date: Wed, 11 Dec 2024 01:53:23 +0100
Subject: [PATCH] Image for `aarch64`

---
 Script/Rocky.aarch64.Dockerfile | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)
 create mode 100644 Script/Rocky.aarch64.Dockerfile

diff --git a/Script/Rocky.aarch64.Dockerfile b/Script/Rocky.aarch64.Dockerfile
new file mode 100644
index 00000000..1000b081
--- /dev/null
+++ b/Script/Rocky.aarch64.Dockerfile
@@ -0,0 +1,32 @@
+FROM arm64v8/rockylinux:9 AS build
+
+RUN dnf install -y epel-release && crb enable
+RUN dnf install -y gcc g++ gfortran cmake wget git hdf5-devel
+
+# part 1: openblas
+RUN git clone --depth 1 --branch v0.3.28 https://github.com/OpenMathLib/OpenBLAS.git
+
+# change the following configurations to match your needs
+RUN cd OpenBLAS && make TARGET=ARMV8 DYNAMIC_ARCH=1 BINARY=64 USE_THREAD=1 USE_OPENMP=1 NUM_THREADS=20 NO_SHARED=1 NO_CBLAS=1 NO_LAPACKE=1 GEMM_MULTITHREAD_THRESHOLD=64
+
+# part 2: tbb
+RUN git clone --depth 1 --branch v2021.12.0 https://github.com/oneapi-src/oneTBB.git
+RUN mkdir tbb-build && cd tbb-build && cmake -DCMAKE_BUILD_TYPE=Release -DTBB_TEST=OFF ../oneTBB && cmake --build . --target install --config Release --parallel "$(nproc)"
+
+# part 3: suanpan
+RUN git clone ---branch dev --depth 1 https://github.com/TLCFEM/suanPan.git
+
+RUN cd suanPan && mkdir build && cd build && cmake -DCMAKE_BUILD_TYPE=Release -DUSE_SYS_LIB=ON -DCUSTOM_OPENBLAS=/OpenBLAS -DBUILD_MULTITHREAD=ON -DUSE_HDF5=ON -DUSE_VTK=OFF -DUSE_AVX2=OFF .. && make -j"$(nproc)"
+
+FROM arm64v8/rockylinux:9 AS runtime
+
+RUN dnf install -y epel-release && crb enable
+RUN dnf install -y libgomp hdf5
+
+COPY --from=build /tbb-build /tbb-build
+
+RUN cd /tbb-build && make install && cd .. && rm -rf tbb-build
+
+COPY --from=build /suanPan/build/suanPan /usr/local/bin/suanPan
+
+RUN suanPan -v