[fix] fix lgbm bytebuffer native order (#3258)

deepjavalibrary · Jun 17, 2024 · 0b768e3 · 0b768e3
1 parent 0b026dc
commit 0b768e3
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/engines/ml/lightgbm/src/main/java/ai/djl/ml/lightgbm/jni/JniUtils.java b/engines/ml/lightgbm/src/main/java/ai/djl/ml/lightgbm/jni/JniUtils.java
@@ -29,6 +29,7 @@
 import com.microsoft.ml.lightgbm.lightgbmlibConstants;
 
 import java.nio.ByteBuffer;
+import java.nio.ByteOrder;
 import java.nio.DoubleBuffer;
 import java.nio.FloatBuffer;
 
@@ -104,15 +105,17 @@ public static Pair<Integer, ByteBuffer> inferenceMat(
             checkCall(result);
             int length = Math.toIntExact(lightgbmlib.int64_tp_value(outLength));
             if (a.getDataType() == DataType.FLOAT32) {
-                ByteBuffer bb = ByteBuffer.allocateDirect(length * 4);
+                ByteBuffer bb =
+                        ByteBuffer.allocateDirect(length * 4).order(ByteOrder.nativeOrder());
                 FloatBuffer wrapped = bb.asFloatBuffer();
                 for (int i = 0; i < length; i++) {
                     wrapped.put((float) lightgbmlib.doubleArray_getitem(outBuffer, i));
                 }
                 bb.rewind();
                 return new Pair<>(length, bb);
             } else if (a.getDataType() == DataType.FLOAT64) {
-                ByteBuffer bb = ByteBuffer.allocateDirect(length * 8);
+                ByteBuffer bb =
+                        ByteBuffer.allocateDirect(length * 8).order(ByteOrder.nativeOrder());
                 DoubleBuffer wrapped = bb.asDoubleBuffer();
                 for (int i = 0; i < length; i++) {
                     wrapped.put(lightgbmlib.doubleArray_getitem(outBuffer, i));