Skip to content

Commit e6f5bd5

Browse files
committed
soy tontoooooooooooo
1 parent 58bd149 commit e6f5bd5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

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

+1-1
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ private static void buildFromTensorFloat(Tensor tensor, String memoryName) throw
117117
ByteBuffer byteBuffer = ByteBuffer.allocateDirect((int) (flatSize * Float.BYTES));
118118
FloatBuffer floatBuffer = byteBuffer.asFloatBuffer();
119119
tensor.data_ptr_float().get(flat);
120-
floatBuffer.put(flatSize);
120+
floatBuffer.put(flat);
121121
byteBuffer.rewind();
122122
shma.getDataBufferNoHeader().put(byteBuffer);
123123
if (PlatformDetection.isWindows()) shma.close();

0 commit comments

Comments
 (0)