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_")) {