Skip to content

Commit a2809d1

Browse files
committed
correct switch
1 parent 88667e5 commit a2809d1

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

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

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,18 +61,24 @@ private ShmBuilder()
6161
*/
6262
public static void build(NDArray tensor, String memoryName) throws IllegalArgumentException, IOException
6363
{
64+
System.out.println(tensor.getDataType().asNumpy());
6465
switch (tensor.getDataType())
6566
{
6667
case UINT8:
6768
buildFromTensorUByte(tensor, memoryName);
69+
break;
6870
case INT32:
6971
buildFromTensorInt(tensor, memoryName);
72+
break;
7073
case FLOAT32:
7174
buildFromTensorFloat(tensor, memoryName);
75+
break;
7276
case FLOAT64:
7377
buildFromTensorDouble(tensor, memoryName);
78+
break;
7479
case INT64:
7580
buildFromTensorLong(tensor, memoryName);
81+
break;
7682
default:
7783
throw new IllegalArgumentException("Unsupported tensor type: " + tensor.getDataType().asNumpy());
7884
}

0 commit comments

Comments
 (0)