[pytorch] Ensure shared library loading order for aarch64

deepjavalibrary · Dec 12, 2023 · 869afe6 · 869afe6
1 parent 3c1301a
commit 869afe6
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_")) {