corerct ultramegabug that was avoiding copying from tensor to shm

Author: carlosuc3m

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

@@ -253,15 +253,18 @@ protected void runFromShmas(List<String> inputs, List<String> outputs) throws IO
         IValue output = model.forward(inputsVector);
         TensorVector outputTensorVector = null;
         if (output.isTensorList()) {
+        	System.out.println("entered 1");
         	outputTensorVector = output.toTensorVector();
         } else {
+        	System.out.println("entered 2");
         	outputTensorVector = new TensorVector();
         	outputTensorVector.put(output.toTensor());
         }
 
 		// Fill the agnostic output tensors list with data from the inference result
 		int c = 0;
 		for (String ee : outputs) {
+			System.out.println(ee);
 			Map<String, Object> decoded = Types.decode(ee);
 			ShmBuilder.build(outputTensorVector.get(c ++), (String) decoded.get(MEM_NAME_KEY));
 		}

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

@@ -25,6 +25,7 @@
 import io.bioimage.modelrunner.utils.CommonUtils;
 
 import java.io.IOException;
+import java.nio.ByteBuffer;
 import java.util.Arrays;
 
 import org.bytedeco.pytorch.Tensor;
@@ -106,7 +107,11 @@ private static void buildFromTensorFloat(Tensor tensor, String memoryName) throw
 			throw new IllegalArgumentException("Model output tensor with shape " + Arrays.toString(arrayShape) 
 					+ " is too big. Max number of elements per float output tensor supported: " + Integer.MAX_VALUE / 4);
 		SharedMemoryArray shma = SharedMemoryArray.readOrCreate(memoryName, arrayShape, new FloatType(), false, true);
-        shma.getDataBufferNoHeader().put(tensor.asByteBuffer());
+		long flatSize = 1;
+    	for (long l : arrayShape) {flatSize *= l;}
+    	ByteBuffer byteBuffer = ByteBuffer.allocate((int) (flatSize * Float.BYTES));
+    	tensor.data_ptr_float().get(byteBuffer.asFloatBuffer().array());
+        shma.getDataBufferNoHeader().put(byteBuffer);
         if (PlatformDetection.isWindows()) shma.close();
     }