Skip to content

Commit

Permalink
Remove calls to inefficient generic Pointer.put()
Browse files Browse the repository at this point in the history
  • Loading branch information
saudet authored and karllessard committed Jan 29, 2020
1 parent 858c7af commit 577994d
Showing 1 changed file with 6 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -939,22 +939,22 @@ private static long write1DArray(Object array, int dtype, BytePointer dst, long
}
switch (dtype) {
case TF_FLOAT:
dst.put(new FloatPointer((float[])array).capacity(nelems));
new FloatPointer(dst).position(dst.position() / 4).put((float[])array);
break;
case TF_DOUBLE:
dst.put(new DoublePointer((double[])array).capacity(nelems));
new DoublePointer(dst).position(dst.position() / 8).put((double[])array);
break;
case TF_INT32:
dst.put(new IntPointer((int[])array).capacity(nelems));
new IntPointer(dst).position(dst.position() / 4).put((int[])array);
break;
case TF_INT64:
dst.put(new LongPointer((long[])array).capacity(nelems));
new LongPointer(dst).position(dst.position() / 8).put((long[])array);
break;
case TF_UINT8:
dst.put(new BytePointer((byte[])array).capacity(nelems));
dst.put((byte[])array);
break;
case TF_BOOL:
dst.put(new BooleanPointer((boolean[])array).capacity(nelems));
new BooleanPointer(dst).position(dst.position()).put((boolean[])array);
break;
default:
throw new IllegalStateException("invalid DataType(" + dtype + ")");
Expand Down

0 comments on commit 577994d

Please sign in to comment.