add more tests

bioimage-io · Nov 23, 2024 · 6486ea4 · 6486ea4
1 parent 292c187
commit 6486ea4
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/shm/ShmBuilder.java b/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/shm/ShmBuilder.java
@@ -103,6 +103,18 @@ private static void buildFromTensorFloat(Tensor tensor, String memoryName) throw
 					+ " is too big. Max number of elements per float output tensor supported: " + Integer.MAX_VALUE / 4);
 		SharedMemoryArray shma = SharedMemoryArray.readOrCreate(memoryName, arrayShape, new UnsignedByteType(), false, true);
         shma.getDataBufferNoHeader().put(tensor.asByteBuffer());
+        shma.getDataBufferNoHeader().rewind();
+        tensor.asByteBuffer().rewind();
+		float sum = 0;
+		while (tensor.asByteBuffer().hasRemaining())
+			sum += tensor.asByteBuffer().getFloat();
+		float sum2 = 0;
+		while (shma.getDataBufferNoHeader().hasRemaining())
+			sum2 += shma.getDataBufferNoHeader().getFloat();
+		System.out.println("KKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKK  SRC  " + sum);
+		System.out.println("KKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKK  TAR  " + sum2);
+        shma.getDataBufferNoHeader().rewind();
+        tensor.asByteBuffer().rewind();
         if (PlatformDetection.isWindows()) shma.close();
     }
 

diff --git a/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/shm/TensorBuilder.java b/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/shm/TensorBuilder.java
@@ -139,10 +139,6 @@ private static org.bytedeco.pytorch.Tensor buildFloat(SharedMemoryArray shmArray
 		float[] flat = new float[buff.capacity() / 4];
 		buff.asFloatBuffer().get(flat);
 		buff.rewind();
-		float sum = 0;
-		for (float ff : flat)
-			sum += ff;
-		System.out.println("KKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKK    " + sum);
 		Tensor ndarray = Tensor.create(flat, ogShape);
 		return ndarray;
 	}