< prev index next >

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

Print this page

        

*** 438,671 **** public static ByteVector random(Species<Byte> s) { ThreadLocalRandom r = ThreadLocalRandom.current(); return ((ByteSpecies)s).op(i -> (byte) r.nextInt()); } - /** - * Returns a mask where each lane is set or unset according to given - * {@code boolean} values - * <p> - * For each mask lane, where {@code N} is the mask lane index, - * if the given {@code boolean} value at index {@code N} is {@code true} - * then the mask lane at index {@code N} is set, otherwise it is unset. - * - * @param species mask species - * @param bits the given {@code boolean} values - * @return a mask where each lane is set or unset according to the given {@code boolean} value - * @throws IndexOutOfBoundsException if {@code bits.length < species.length()} - */ - @ForceInline - public static Mask<Byte> maskFromValues(Species<Byte> species, boolean... bits) { - if (species.boxType() == ByteMaxVector.class) - return new ByteMaxVector.ByteMaxMask(bits); - switch (species.bitSize()) { - case 64: return new Byte64Vector.Byte64Mask(bits); - case 128: return new Byte128Vector.Byte128Mask(bits); - case 256: return new Byte256Vector.Byte256Mask(bits); - case 512: return new Byte512Vector.Byte512Mask(bits); - default: throw new IllegalArgumentException(Integer.toString(species.bitSize())); - } - } - - // @@@ This is a bad implementation -- makes lambdas capturing -- fix this - static Mask<Byte> trueMask(Species<Byte> species) { - if (species.boxType() == ByteMaxVector.class) - return ByteMaxVector.ByteMaxMask.TRUE_MASK; - switch (species.bitSize()) { - case 64: return Byte64Vector.Byte64Mask.TRUE_MASK; - case 128: return Byte128Vector.Byte128Mask.TRUE_MASK; - case 256: return Byte256Vector.Byte256Mask.TRUE_MASK; - case 512: return Byte512Vector.Byte512Mask.TRUE_MASK; - default: throw new IllegalArgumentException(Integer.toString(species.bitSize())); - } - } - - static Mask<Byte> falseMask(Species<Byte> species) { - if (species.boxType() == ByteMaxVector.class) - return ByteMaxVector.ByteMaxMask.FALSE_MASK; - switch (species.bitSize()) { - case 64: return Byte64Vector.Byte64Mask.FALSE_MASK; - case 128: return Byte128Vector.Byte128Mask.FALSE_MASK; - case 256: return Byte256Vector.Byte256Mask.FALSE_MASK; - case 512: return Byte512Vector.Byte512Mask.FALSE_MASK; - default: throw new IllegalArgumentException(Integer.toString(species.bitSize())); - } - } - - /** - * Loads a mask from a {@code boolean} array starting at an offset. - * <p> - * For each mask lane, where {@code N} is the mask lane index, - * if the array element at index {@code ix + N} is {@code true} then the - * mask lane at index {@code N} is set, otherwise it is unset. - * - * @param species mask species - * @param bits the {@code boolean} array - * @param ix the offset into the array - * @return the mask loaded from a {@code boolean} array - * @throws IndexOutOfBoundsException if {@code ix < 0}, or - * {@code ix > bits.length - species.length()} - */ - @ForceInline - @SuppressWarnings("unchecked") - public static Mask<Byte> maskFromArray(Species<Byte> species, boolean[] bits, int ix) { - Objects.requireNonNull(bits); - ix = VectorIntrinsics.checkIndex(ix, bits.length, species.length()); - return VectorIntrinsics.load((Class<Mask<Byte>>) species.maskType(), byte.class, species.length(), - bits, (((long) ix) << ARRAY_SHIFT) + Unsafe.ARRAY_BOOLEAN_BASE_OFFSET, - bits, ix, species, - (c, idx, s) -> (Mask<Byte>) ((ByteSpecies)s).opm(n -> c[idx + n])); - } - - /** - * Returns a mask where all lanes are set. - * - * @param species mask species - * @return a mask where all lanes are set - */ - @ForceInline - @SuppressWarnings("unchecked") - public static Mask<Byte> maskAllTrue(Species<Byte> species) { - return VectorIntrinsics.broadcastCoerced((Class<Mask<Byte>>) species.maskType(), byte.class, species.length(), - (byte)-1, species, - ((z, s) -> trueMask(s))); - } - - /** - * Returns a mask where all lanes are unset. - * - * @param species mask species - * @return a mask where all lanes are unset - */ - @ForceInline - @SuppressWarnings("unchecked") - public static Mask<Byte> maskAllFalse(Species<Byte> species) { - return VectorIntrinsics.broadcastCoerced((Class<Mask<Byte>>) species.maskType(), byte.class, species.length(), - 0, species, - ((z, s) -> falseMask(s))); - } - - /** - * Returns a shuffle of mapped indexes where each lane element is - * the result of applying a mapping function to the corresponding lane - * index. - * <p> - * Care should be taken to ensure Shuffle values produced from this - * method are consumed as constants to ensure optimal generation of - * code. For example, values held in static final fields or values - * held in loop constant local variables. - * <p> - * This method behaves as if a shuffle is created from an array of - * mapped indexes as follows: - * <pre>{@code - * int[] a = new int[species.length()]; - * for (int i = 0; i < a.length; i++) { - * a[i] = f.applyAsInt(i); - * } - * return this.shuffleFromValues(a); - * }</pre> - * - * @param species shuffle species - * @param f the lane index mapping function - * @return a shuffle of mapped indexes - */ - @ForceInline - public static Shuffle<Byte> shuffle(Species<Byte> species, IntUnaryOperator f) { - if (species.boxType() == ByteMaxVector.class) - return new ByteMaxVector.ByteMaxShuffle(f); - switch (species.bitSize()) { - case 64: return new Byte64Vector.Byte64Shuffle(f); - case 128: return new Byte128Vector.Byte128Shuffle(f); - case 256: return new Byte256Vector.Byte256Shuffle(f); - case 512: return new Byte512Vector.Byte512Shuffle(f); - default: throw new IllegalArgumentException(Integer.toString(species.bitSize())); - } - } - - /** - * Returns a shuffle where each lane element is the value of its - * corresponding lane index. - * <p> - * This method behaves as if a shuffle is created from an identity - * index mapping function as follows: - * <pre>{@code - * return this.shuffle(i -> i); - * }</pre> - * - * @param species shuffle species - * @return a shuffle of lane indexes - */ - @ForceInline - public static Shuffle<Byte> shuffleIota(Species<Byte> species) { - if (species.boxType() == ByteMaxVector.class) - return new ByteMaxVector.ByteMaxShuffle(AbstractShuffle.IDENTITY); - switch (species.bitSize()) { - case 64: return new Byte64Vector.Byte64Shuffle(AbstractShuffle.IDENTITY); - case 128: return new Byte128Vector.Byte128Shuffle(AbstractShuffle.IDENTITY); - case 256: return new Byte256Vector.Byte256Shuffle(AbstractShuffle.IDENTITY); - case 512: return new Byte512Vector.Byte512Shuffle(AbstractShuffle.IDENTITY); - default: throw new IllegalArgumentException(Integer.toString(species.bitSize())); - } - } - - /** - * Returns a shuffle where each lane element is set to a given - * {@code int} value logically AND'ed by the species length minus one. - * <p> - * For each shuffle lane, where {@code N} is the shuffle lane index, the - * the {@code int} value at index {@code N} logically AND'ed by - * {@code species.length() - 1} is placed into the resulting shuffle at - * lane index {@code N}. - * - * @param species shuffle species - * @param ixs the given {@code int} values - * @return a shuffle where each lane element is set to a given - * {@code int} value - * @throws IndexOutOfBoundsException if the number of int values is - * {@code < species.length()} - */ - @ForceInline - public static Shuffle<Byte> shuffleFromValues(Species<Byte> species, int... ixs) { - if (species.boxType() == ByteMaxVector.class) - return new ByteMaxVector.ByteMaxShuffle(ixs); - switch (species.bitSize()) { - case 64: return new Byte64Vector.Byte64Shuffle(ixs); - case 128: return new Byte128Vector.Byte128Shuffle(ixs); - case 256: return new Byte256Vector.Byte256Shuffle(ixs); - case 512: return new Byte512Vector.Byte512Shuffle(ixs); - default: throw new IllegalArgumentException(Integer.toString(species.bitSize())); - } - } - - /** - * Loads a shuffle from an {@code int} array starting at an offset. - * <p> - * For each shuffle lane, where {@code N} is the shuffle lane index, the - * array element at index {@code i + N} logically AND'ed by - * {@code species.length() - 1} is placed into the resulting shuffle at lane - * index {@code N}. - * - * @param species shuffle species - * @param ixs the {@code int} array - * @param i the offset into the array - * @return a shuffle loaded from the {@code int} array - * @throws IndexOutOfBoundsException if {@code i < 0}, or - * {@code i > a.length - species.length()} - */ - @ForceInline - public static Shuffle<Byte> shuffleFromArray(Species<Byte> species, int[] ixs, int i) { - if (species.boxType() == ByteMaxVector.class) - return new ByteMaxVector.ByteMaxShuffle(ixs, i); - switch (species.bitSize()) { - case 64: return new Byte64Vector.Byte64Shuffle(ixs, i); - case 128: return new Byte128Vector.Byte128Shuffle(ixs, i); - case 256: return new Byte256Vector.Byte256Shuffle(ixs, i); - case 512: return new Byte512Vector.Byte512Shuffle(ixs, i); - default: throw new IllegalArgumentException(Integer.toString(species.bitSize())); - } - } - // Ops @Override public abstract ByteVector add(Vector<Byte> v); --- 438,447 ----
*** 1545,1568 **** @Override public abstract Species<Byte> species(); /** ! * Class representing {@link ByteVector}'s of the same {@link Vector.Shape Shape}. */ ! static final class ByteSpecies extends Vector.AbstractSpecies<Byte> { final Function<byte[], ByteVector> vectorFactory; - final Function<boolean[], Vector.Mask<Byte>> maskFactory; ! private ByteSpecies(Vector.Shape shape, Class<?> boxType, Class<?> maskType, Function<byte[], ByteVector> vectorFactory, ! Function<boolean[], Vector.Mask<Byte>> maskFactory) { ! super(shape, byte.class, Byte.SIZE, boxType, maskType); this.vectorFactory = vectorFactory; - this.maskFactory = maskFactory; } interface FOp { byte apply(int i); } --- 1321,1345 ---- @Override public abstract Species<Byte> species(); /** ! * Class representing {@link ByteVector}'s of the same {@link Shape Shape}. */ ! static final class ByteSpecies extends AbstractSpecies<Byte> { final Function<byte[], ByteVector> vectorFactory; ! private ByteSpecies(Shape shape, Class<?> boxType, Class<?> maskType, Function<byte[], ByteVector> vectorFactory, ! Function<boolean[], Mask<Byte>> maskFactory, ! Function<IntUnaryOperator, Shuffle<Byte>> shuffleFromArrayFactory, ! fShuffleFromArray<Byte> shuffleFromOpFactory) { ! super(shape, byte.class, Byte.SIZE, boxType, maskType, maskFactory, ! shuffleFromArrayFactory, shuffleFromOpFactory); this.vectorFactory = vectorFactory; } interface FOp { byte apply(int i); }
*** 1577,1604 **** res[i] = f.apply(i); } return vectorFactory.apply(res); } ! ByteVector op(Vector.Mask<Byte> o, FOp f) { byte[] res = new byte[length()]; boolean[] mbits = ((AbstractMask<Byte>)o).getBits(); for (int i = 0; i < length(); i++) { if (mbits[i]) { res[i] = f.apply(i); } } return vectorFactory.apply(res); } - - Vector.Mask<Byte> opm(IntVector.IntSpecies.FOpm f) { - boolean[] res = new boolean[length()]; - for (int i = 0; i < length(); i++) { - res[i] = (boolean)f.apply(i); - } - return maskFactory.apply(res); - } } /** * Finds the preferred species for an element type of {@code byte}. * <p> --- 1354,1373 ---- res[i] = f.apply(i); } return vectorFactory.apply(res); } ! ByteVector op(Mask<Byte> o, FOp f) { byte[] res = new byte[length()]; boolean[] mbits = ((AbstractMask<Byte>)o).getBits(); for (int i = 0; i < length(); i++) { if (mbits[i]) { res[i] = f.apply(i); } } return vectorFactory.apply(res); } } /** * Finds the preferred species for an element type of {@code byte}. * <p>
*** 1618,1628 **** * * @param s the shape * @return a species for an element type of {@code byte} and shape * @throws IllegalArgumentException if no such species exists for the shape */ ! static ByteSpecies species(Vector.Shape s) { Objects.requireNonNull(s); switch (s) { case S_64_BIT: return (ByteSpecies) SPECIES_64; case S_128_BIT: return (ByteSpecies) SPECIES_128; case S_256_BIT: return (ByteSpecies) SPECIES_256; --- 1387,1397 ---- * * @param s the shape * @return a species for an element type of {@code byte} and shape * @throws IllegalArgumentException if no such species exists for the shape */ ! static ByteSpecies species(Shape s) { Objects.requireNonNull(s); switch (s) { case S_64_BIT: return (ByteSpecies) SPECIES_64; case S_128_BIT: return (ByteSpecies) SPECIES_128; case S_256_BIT: return (ByteSpecies) SPECIES_256;
*** 1630,1658 **** case S_Max_BIT: return (ByteSpecies) SPECIES_MAX; default: throw new IllegalArgumentException("Bad shape: " + s); } } ! /** Species representing {@link ByteVector}s of {@link Vector.Shape#S_64_BIT Shape.S_64_BIT}. */ public static final Species<Byte> SPECIES_64 = new ByteSpecies(Shape.S_64_BIT, Byte64Vector.class, Byte64Vector.Byte64Mask.class, ! Byte64Vector::new, Byte64Vector.Byte64Mask::new); ! /** Species representing {@link ByteVector}s of {@link Vector.Shape#S_128_BIT Shape.S_128_BIT}. */ public static final Species<Byte> SPECIES_128 = new ByteSpecies(Shape.S_128_BIT, Byte128Vector.class, Byte128Vector.Byte128Mask.class, ! Byte128Vector::new, Byte128Vector.Byte128Mask::new); ! /** Species representing {@link ByteVector}s of {@link Vector.Shape#S_256_BIT Shape.S_256_BIT}. */ public static final Species<Byte> SPECIES_256 = new ByteSpecies(Shape.S_256_BIT, Byte256Vector.class, Byte256Vector.Byte256Mask.class, ! Byte256Vector::new, Byte256Vector.Byte256Mask::new); ! /** Species representing {@link ByteVector}s of {@link Vector.Shape#S_512_BIT Shape.S_512_BIT}. */ public static final Species<Byte> SPECIES_512 = new ByteSpecies(Shape.S_512_BIT, Byte512Vector.class, Byte512Vector.Byte512Mask.class, ! Byte512Vector::new, Byte512Vector.Byte512Mask::new); ! /** Species representing {@link ByteVector}s of {@link Vector.Shape#S_Max_BIT Shape.S_Max_BIT}. */ public static final Species<Byte> SPECIES_MAX = new ByteSpecies(Shape.S_Max_BIT, ByteMaxVector.class, ByteMaxVector.ByteMaxMask.class, ! ByteMaxVector::new, ByteMaxVector.ByteMaxMask::new); /** * Preferred species for {@link ByteVector}s. * A preferred species is a species of maximal bit size for the platform. */ --- 1399,1432 ---- case S_Max_BIT: return (ByteSpecies) SPECIES_MAX; default: throw new IllegalArgumentException("Bad shape: " + s); } } ! /** Species representing {@link ByteVector}s of {@link Shape#S_64_BIT Shape.S_64_BIT}. */ public static final Species<Byte> SPECIES_64 = new ByteSpecies(Shape.S_64_BIT, Byte64Vector.class, Byte64Vector.Byte64Mask.class, ! Byte64Vector::new, Byte64Vector.Byte64Mask::new, ! Byte64Vector.Byte64Shuffle::new, Byte64Vector.Byte64Shuffle::new); ! /** Species representing {@link ByteVector}s of {@link Shape#S_128_BIT Shape.S_128_BIT}. */ public static final Species<Byte> SPECIES_128 = new ByteSpecies(Shape.S_128_BIT, Byte128Vector.class, Byte128Vector.Byte128Mask.class, ! Byte128Vector::new, Byte128Vector.Byte128Mask::new, ! Byte128Vector.Byte128Shuffle::new, Byte128Vector.Byte128Shuffle::new); ! /** Species representing {@link ByteVector}s of {@link Shape#S_256_BIT Shape.S_256_BIT}. */ public static final Species<Byte> SPECIES_256 = new ByteSpecies(Shape.S_256_BIT, Byte256Vector.class, Byte256Vector.Byte256Mask.class, ! Byte256Vector::new, Byte256Vector.Byte256Mask::new, ! Byte256Vector.Byte256Shuffle::new, Byte256Vector.Byte256Shuffle::new); ! /** Species representing {@link ByteVector}s of {@link Shape#S_512_BIT Shape.S_512_BIT}. */ public static final Species<Byte> SPECIES_512 = new ByteSpecies(Shape.S_512_BIT, Byte512Vector.class, Byte512Vector.Byte512Mask.class, ! Byte512Vector::new, Byte512Vector.Byte512Mask::new, ! Byte512Vector.Byte512Shuffle::new, Byte512Vector.Byte512Shuffle::new); ! /** Species representing {@link ByteVector}s of {@link Shape#S_Max_BIT Shape.S_Max_BIT}. */ public static final Species<Byte> SPECIES_MAX = new ByteSpecies(Shape.S_Max_BIT, ByteMaxVector.class, ByteMaxVector.ByteMaxMask.class, ! ByteMaxVector::new, ByteMaxVector.ByteMaxMask::new, ! ByteMaxVector.ByteMaxShuffle::new, ByteMaxVector.ByteMaxShuffle::new); /** * Preferred species for {@link ByteVector}s. * A preferred species is a species of maximal bit size for the platform. */
< prev index next >