From 869afe633ddee61773cbba00a583b4efe7208d90 Mon Sep 17 00:00:00 2001
From: Frank Liu <frankfliu2000@gmail.com>
Date: Tue, 12 Dec 2023 13:28:58 -0800
Subject: [PATCH] [pytorch] Ensure shared library loading order for aarch64

---
 .../src/main/java/ai/djl/pytorch/jni/LibUtils.java             | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/engines/pytorch/pytorch-engine/src/main/java/ai/djl/pytorch/jni/LibUtils.java b/engines/pytorch/pytorch-engine/src/main/java/ai/djl/pytorch/jni/LibUtils.java
index 860c1bc48e8..b28a7a9d19a 100644
--- a/engines/pytorch/pytorch-engine/src/main/java/ai/djl/pytorch/jni/LibUtils.java
+++ b/engines/pytorch/pytorch-engine/src/main/java/ai/djl/pytorch/jni/LibUtils.java
@@ -130,7 +130,8 @@ private static void loadLibTorch(LibTorch libTorch) {
                                         && name.contains("cudart")
                                         && name.contains("nvTools")) {
                                     return false;
-                                } else if (name.startsWith("libarm_compute-")) {
+                                } else if (name.startsWith("libarm_compute-")
+                                        || name.startsWith("libopenblasp")) {
                                     rank.put(path, 2);
                                     return true;
                                 } else if (name.startsWith("libarm_compute_")) {