< prev index next >

src/jdk.incubator.vector/share/classes/jdk/incubator/vector/Float512Vector.java

Print this page
rev 49509 : [vector] Intrinsic support for resize

*** 717,729 **** public <F> Vector<F, Shapes.S512Bit> rebracket(Species<F, Shapes.S512Bit> species) { Objects.requireNonNull(species); // TODO: check proper element type // TODO: update to pass the two species as an arguments and ideally // push down intrinsic call into species implementation ! return VectorIntrinsics.rebracket( Float512Vector.class, float.class, LENGTH, ! species.elementType(), this, (v, t) -> species.reshape(v) ); } // Accessors --- 717,729 ---- public <F> Vector<F, Shapes.S512Bit> rebracket(Species<F, Shapes.S512Bit> species) { Objects.requireNonNull(species); // TODO: check proper element type // TODO: update to pass the two species as an arguments and ideally // push down intrinsic call into species implementation ! return VectorIntrinsics.reinterpret( Float512Vector.class, float.class, LENGTH, ! species.elementType(), species.length(), this, (v, t) -> species.reshape(v) ); } // Accessors
*** 808,820 **** @ForceInline @SuppressWarnings("unchecked") public <Z> Mask<Z, Shapes.S512Bit> rebracket(Species<Z, Shapes.S512Bit> species) { Objects.requireNonNull(species); // TODO: check proper element type ! return VectorIntrinsics.rebracket( Float512Mask.class, float.class, LENGTH, ! species.elementType(), this, (m, t) -> m.reshape(species) ); } // Unary operations --- 808,820 ---- @ForceInline @SuppressWarnings("unchecked") public <Z> Mask<Z, Shapes.S512Bit> rebracket(Species<Z, Shapes.S512Bit> species) { Objects.requireNonNull(species); // TODO: check proper element type ! return VectorIntrinsics.reinterpret( Float512Mask.class, float.class, LENGTH, ! species.elementType(), species.length(), this, (m, t) -> m.reshape(species) ); } // Unary operations
*** 907,936 **** --- 907,941 ---- sb.append("]"); return sb.toString(); } @Override + @ForceInline public int bitSize() { return BIT_SIZE; } @Override + @ForceInline public int length() { return LENGTH; } @Override + @ForceInline public Class<Float> elementType() { return Float.class; } @Override + @ForceInline public int elementSize() { return Float.SIZE; } @Override + @ForceInline public Shapes.S512Bit shape() { return Shapes.S_512_BIT; } @Override
*** 1022,1028 **** --- 1027,1071 ---- @Override @ForceInline public Float512Vector fromArray(float[] a, int ax, Mask<Float, Shapes.S512Bit> m) { return zero().blend(fromArray(a, ax), m); // TODO: use better default impl: op(m, i -> a[ax + i]); } + + @Override + @ForceInline + @SuppressWarnings("unchecked") + public <T extends Shape> Float512Vector resize(Vector<Float, T> o) { + Objects.requireNonNull(o); + if (o.bitSize() == 64) { + Float64Vector so = (Float64Vector)o; + return VectorIntrinsics.reinterpret( + Float64Vector.class, float.class, so.length(), + Float.class, LENGTH, so, + (v, t) -> (Float512Vector)reshape(v) + ); + } else if (o.bitSize() == 128) { + Float128Vector so = (Float128Vector)o; + return VectorIntrinsics.reinterpret( + Float128Vector.class, float.class, so.length(), + Float.class, LENGTH, so, + (v, t) -> (Float512Vector)reshape(v) + ); + } else if (o.bitSize() == 256) { + Float256Vector so = (Float256Vector)o; + return VectorIntrinsics.reinterpret( + Float256Vector.class, float.class, so.length(), + Float.class, LENGTH, so, + (v, t) -> (Float512Vector)reshape(v) + ); + } else if (o.bitSize() == 512) { + Float512Vector so = (Float512Vector)o; + return VectorIntrinsics.reinterpret( + Float512Vector.class, float.class, so.length(), + Float.class, LENGTH, so, + (v, t) -> (Float512Vector)reshape(v) + ); + } else { + throw new InternalError("Unimplemented size"); + } + } } }
< prev index next >