Skip to content

Commit 2a12088

Browse files
committed
tiro en los wevos
1 parent 76c8143 commit 2a12088

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

Diff for: src/main/java/io/bioimage/modelrunner/pytorch/javacpp/shm/ShmBuilder.java

+2-1
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@
2626

2727
import java.io.IOException;
2828
import java.nio.ByteBuffer;
29+
import java.nio.ByteOrder;
2930
import java.nio.DoubleBuffer;
3031
import java.nio.FloatBuffer;
3132
import java.nio.IntBuffer;
@@ -134,7 +135,7 @@ private static void buildFromTensorFloat(Tensor tensor, String memoryName) throw
134135
long flatSize = 1;
135136
for (long l : arrayShape) {flatSize *= l;}
136137
float[] flat = new float[(int) flatSize];
137-
ByteBuffer byteBuffer = ByteBuffer.allocateDirect((int) (flatSize * Float.BYTES));
138+
ByteBuffer byteBuffer = ByteBuffer.allocateDirect((int) (flatSize * Float.BYTES)).order(ByteOrder.LITTLE_ENDIAN);
138139
FloatBuffer floatBuffer = byteBuffer.asFloatBuffer();
139140
tensor.data_ptr_float().get(flat);
140141
floatBuffer.put(flat);

0 commit comments

Comments
 (0)