linux/aarch64 binary

metarank · Nov 16, 2022 · fae850f · fae850f
1 parent d515464
commit fae850f
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 5 deletions.
diff --git a/pom.xml b/pom.xml
@@ -6,7 +6,7 @@
 
     <groupId>io.github.metarank</groupId>
     <artifactId>lightgbm4j</artifactId>
-    <version>3.3.2-2</version>
+    <version>3.3.2-3</version>
 
     <name>LightGBM4j: Java LightGBM wrapper</name>
     <description>A high-level wrapper for LightGBM toolkit</description>

diff --git a/src/main/java/io/github/metarank/lightgbm4j/LGBMBooster.java b/src/main/java/io/github/metarank/lightgbm4j/LGBMBooster.java
@@ -43,12 +43,18 @@ public static boolean isNativeLoaded() {
     public synchronized static void loadNative() throws IOException {
         if (!nativeLoaded) {
             String os = System.getProperty("os.name");
+            String arch = System.getProperty("os.arch", "generic").toLowerCase(Locale.ENGLISH);
             if (os.startsWith("Linux") || os.startsWith("LINUX")) {
-                loadNative("linux/x86_64/lib_lightgbm.so", "lib_lightgbm.so");
-                loadNative("linux/x86_64/lib_lightgbm_swig.so", "lib_lightgbm_swig.so");
-                nativeLoaded = true;
+                if (arch.startsWith("amd64") || arch.startsWith("x86_64")) {
+                    loadNative("linux/x86_64/lib_lightgbm.so", "lib_lightgbm.so");
+                    loadNative("linux/x86_64/lib_lightgbm_swig.so", "lib_lightgbm_swig.so");
+                    nativeLoaded = true;
+                } else if (arch.startsWith("aarch64") || arch.startsWith("arm64")) {
+                    loadNative("linux/aarch64/lib_lightgbm.so", "lib_lightgbm.so");
+                    loadNative("linux/aarch64/lib_lightgbm_swig.so", "lib_lightgbm_swig.so");
+                    nativeLoaded = true;
+                }
             } else if (os.startsWith("Mac")) {
-                String arch = System.getProperty("os.arch", "generic").toLowerCase(Locale.ENGLISH);
                 if (arch.startsWith("amd64") || arch.startsWith("x86_64")) {
                     loadNative("osx/x86_64/lib_lightgbm.dylib", "lib_lightgbm.dylib");
                     loadNative("osx/x86_64/lib_lightgbm_swig.dylib", "lib_lightgbm_swig.dylib");

diff --git a/src/main/resources/linux/aarch64/lib_lightgbm.so b/src/main/resources/linux/aarch64/lib_lightgbm.so
diff --git a/src/main/resources/linux/aarch64/lib_lightgbm_swig.so b/src/main/resources/linux/aarch64/lib_lightgbm_swig.so