[pytorch] Ensure shared library loading order for aarch64 (#2892)

deepjavalibrary · Apr 26, 2024 · 1a47bea · 1a47bea
1 parent 66a106a
commit 1a47bea
Showing 1 changed file with 2 additions and 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
@@ -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_")) {