< prev index next >
src/jdk.incubator.vector/share/classes/jdk/incubator/vector/LongVector.java
Print this page
*** 462,695 ****
public static LongVector random(Species<Long> s) {
ThreadLocalRandom r = ThreadLocalRandom.current();
return ((LongSpecies)s).op(i -> r.nextLong());
}
- /**
- * 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<Long> maskFromValues(Species<Long> species, boolean... bits) {
- if (species.boxType() == LongMaxVector.class)
- return new LongMaxVector.LongMaxMask(bits);
- switch (species.bitSize()) {
- case 64: return new Long64Vector.Long64Mask(bits);
- case 128: return new Long128Vector.Long128Mask(bits);
- case 256: return new Long256Vector.Long256Mask(bits);
- case 512: return new Long512Vector.Long512Mask(bits);
- default: throw new IllegalArgumentException(Integer.toString(species.bitSize()));
- }
- }
-
- // @@@ This is a bad implementation -- makes lambdas capturing -- fix this
- static Mask<Long> trueMask(Species<Long> species) {
- if (species.boxType() == LongMaxVector.class)
- return LongMaxVector.LongMaxMask.TRUE_MASK;
- switch (species.bitSize()) {
- case 64: return Long64Vector.Long64Mask.TRUE_MASK;
- case 128: return Long128Vector.Long128Mask.TRUE_MASK;
- case 256: return Long256Vector.Long256Mask.TRUE_MASK;
- case 512: return Long512Vector.Long512Mask.TRUE_MASK;
- default: throw new IllegalArgumentException(Integer.toString(species.bitSize()));
- }
- }
-
- static Mask<Long> falseMask(Species<Long> species) {
- if (species.boxType() == LongMaxVector.class)
- return LongMaxVector.LongMaxMask.FALSE_MASK;
- switch (species.bitSize()) {
- case 64: return Long64Vector.Long64Mask.FALSE_MASK;
- case 128: return Long128Vector.Long128Mask.FALSE_MASK;
- case 256: return Long256Vector.Long256Mask.FALSE_MASK;
- case 512: return Long512Vector.Long512Mask.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<Long> maskFromArray(Species<Long> species, boolean[] bits, int ix) {
- Objects.requireNonNull(bits);
- ix = VectorIntrinsics.checkIndex(ix, bits.length, species.length());
- return VectorIntrinsics.load((Class<Mask<Long>>) species.maskType(), long.class, species.length(),
- bits, (((long) ix) << ARRAY_SHIFT) + Unsafe.ARRAY_BOOLEAN_BASE_OFFSET,
- bits, ix, species,
- (c, idx, s) -> (Mask<Long>) ((LongSpecies)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<Long> maskAllTrue(Species<Long> species) {
- return VectorIntrinsics.broadcastCoerced((Class<Mask<Long>>) species.maskType(), long.class, species.length(),
- (long)-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<Long> maskAllFalse(Species<Long> species) {
- return VectorIntrinsics.broadcastCoerced((Class<Mask<Long>>) species.maskType(), long.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<Long> shuffle(Species<Long> species, IntUnaryOperator f) {
- if (species.boxType() == LongMaxVector.class)
- return new LongMaxVector.LongMaxShuffle(f);
- switch (species.bitSize()) {
- case 64: return new Long64Vector.Long64Shuffle(f);
- case 128: return new Long128Vector.Long128Shuffle(f);
- case 256: return new Long256Vector.Long256Shuffle(f);
- case 512: return new Long512Vector.Long512Shuffle(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<Long> shuffleIota(Species<Long> species) {
- if (species.boxType() == LongMaxVector.class)
- return new LongMaxVector.LongMaxShuffle(AbstractShuffle.IDENTITY);
- switch (species.bitSize()) {
- case 64: return new Long64Vector.Long64Shuffle(AbstractShuffle.IDENTITY);
- case 128: return new Long128Vector.Long128Shuffle(AbstractShuffle.IDENTITY);
- case 256: return new Long256Vector.Long256Shuffle(AbstractShuffle.IDENTITY);
- case 512: return new Long512Vector.Long512Shuffle(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<Long> shuffleFromValues(Species<Long> species, int... ixs) {
- if (species.boxType() == LongMaxVector.class)
- return new LongMaxVector.LongMaxShuffle(ixs);
- switch (species.bitSize()) {
- case 64: return new Long64Vector.Long64Shuffle(ixs);
- case 128: return new Long128Vector.Long128Shuffle(ixs);
- case 256: return new Long256Vector.Long256Shuffle(ixs);
- case 512: return new Long512Vector.Long512Shuffle(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<Long> shuffleFromArray(Species<Long> species, int[] ixs, int i) {
- if (species.boxType() == LongMaxVector.class)
- return new LongMaxVector.LongMaxShuffle(ixs, i);
- switch (species.bitSize()) {
- case 64: return new Long64Vector.Long64Shuffle(ixs, i);
- case 128: return new Long128Vector.Long128Shuffle(ixs, i);
- case 256: return new Long256Vector.Long256Shuffle(ixs, i);
- case 512: return new Long512Vector.Long512Shuffle(ixs, i);
- default: throw new IllegalArgumentException(Integer.toString(species.bitSize()));
- }
- }
-
// Ops
@Override
public abstract LongVector add(Vector<Long> v);
--- 462,471 ----
*** 1700,1723 ****
@Override
public abstract Species<Long> species();
/**
! * Class representing {@link LongVector}'s of the same {@link Vector.Shape Shape}.
*/
! static final class LongSpecies extends Vector.AbstractSpecies<Long> {
final Function<long[], LongVector> vectorFactory;
- final Function<boolean[], Vector.Mask<Long>> maskFactory;
! private LongSpecies(Vector.Shape shape,
Class<?> boxType,
Class<?> maskType,
Function<long[], LongVector> vectorFactory,
! Function<boolean[], Vector.Mask<Long>> maskFactory) {
! super(shape, long.class, Long.SIZE, boxType, maskType);
this.vectorFactory = vectorFactory;
- this.maskFactory = maskFactory;
}
interface FOp {
long apply(int i);
}
--- 1476,1500 ----
@Override
public abstract Species<Long> species();
/**
! * Class representing {@link LongVector}'s of the same {@link Shape Shape}.
*/
! static final class LongSpecies extends AbstractSpecies<Long> {
final Function<long[], LongVector> vectorFactory;
! private LongSpecies(Shape shape,
Class<?> boxType,
Class<?> maskType,
Function<long[], LongVector> vectorFactory,
! Function<boolean[], Mask<Long>> maskFactory,
! Function<IntUnaryOperator, Shuffle<Long>> shuffleFromArrayFactory,
! fShuffleFromArray<Long> shuffleFromOpFactory) {
! super(shape, long.class, Long.SIZE, boxType, maskType, maskFactory,
! shuffleFromArrayFactory, shuffleFromOpFactory);
this.vectorFactory = vectorFactory;
}
interface FOp {
long apply(int i);
}
*** 1732,1759 ****
res[i] = f.apply(i);
}
return vectorFactory.apply(res);
}
! LongVector op(Vector.Mask<Long> o, FOp f) {
long[] res = new long[length()];
boolean[] mbits = ((AbstractMask<Long>)o).getBits();
for (int i = 0; i < length(); i++) {
if (mbits[i]) {
res[i] = f.apply(i);
}
}
return vectorFactory.apply(res);
}
-
- Vector.Mask<Long> 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 long}.
* <p>
--- 1509,1528 ----
res[i] = f.apply(i);
}
return vectorFactory.apply(res);
}
! LongVector op(Mask<Long> o, FOp f) {
long[] res = new long[length()];
boolean[] mbits = ((AbstractMask<Long>)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 long}.
* <p>
*** 1773,1783 ****
*
* @param s the shape
* @return a species for an element type of {@code long} and shape
* @throws IllegalArgumentException if no such species exists for the shape
*/
! static LongSpecies species(Vector.Shape s) {
Objects.requireNonNull(s);
switch (s) {
case S_64_BIT: return (LongSpecies) SPECIES_64;
case S_128_BIT: return (LongSpecies) SPECIES_128;
case S_256_BIT: return (LongSpecies) SPECIES_256;
--- 1542,1552 ----
*
* @param s the shape
* @return a species for an element type of {@code long} and shape
* @throws IllegalArgumentException if no such species exists for the shape
*/
! static LongSpecies species(Shape s) {
Objects.requireNonNull(s);
switch (s) {
case S_64_BIT: return (LongSpecies) SPECIES_64;
case S_128_BIT: return (LongSpecies) SPECIES_128;
case S_256_BIT: return (LongSpecies) SPECIES_256;
*** 1785,1813 ****
case S_Max_BIT: return (LongSpecies) SPECIES_MAX;
default: throw new IllegalArgumentException("Bad shape: " + s);
}
}
! /** Species representing {@link LongVector}s of {@link Vector.Shape#S_64_BIT Shape.S_64_BIT}. */
public static final Species<Long> SPECIES_64 = new LongSpecies(Shape.S_64_BIT, Long64Vector.class, Long64Vector.Long64Mask.class,
! Long64Vector::new, Long64Vector.Long64Mask::new);
! /** Species representing {@link LongVector}s of {@link Vector.Shape#S_128_BIT Shape.S_128_BIT}. */
public static final Species<Long> SPECIES_128 = new LongSpecies(Shape.S_128_BIT, Long128Vector.class, Long128Vector.Long128Mask.class,
! Long128Vector::new, Long128Vector.Long128Mask::new);
! /** Species representing {@link LongVector}s of {@link Vector.Shape#S_256_BIT Shape.S_256_BIT}. */
public static final Species<Long> SPECIES_256 = new LongSpecies(Shape.S_256_BIT, Long256Vector.class, Long256Vector.Long256Mask.class,
! Long256Vector::new, Long256Vector.Long256Mask::new);
! /** Species representing {@link LongVector}s of {@link Vector.Shape#S_512_BIT Shape.S_512_BIT}. */
public static final Species<Long> SPECIES_512 = new LongSpecies(Shape.S_512_BIT, Long512Vector.class, Long512Vector.Long512Mask.class,
! Long512Vector::new, Long512Vector.Long512Mask::new);
! /** Species representing {@link LongVector}s of {@link Vector.Shape#S_Max_BIT Shape.S_Max_BIT}. */
public static final Species<Long> SPECIES_MAX = new LongSpecies(Shape.S_Max_BIT, LongMaxVector.class, LongMaxVector.LongMaxMask.class,
! LongMaxVector::new, LongMaxVector.LongMaxMask::new);
/**
* Preferred species for {@link LongVector}s.
* A preferred species is a species of maximal bit size for the platform.
*/
--- 1554,1587 ----
case S_Max_BIT: return (LongSpecies) SPECIES_MAX;
default: throw new IllegalArgumentException("Bad shape: " + s);
}
}
! /** Species representing {@link LongVector}s of {@link Shape#S_64_BIT Shape.S_64_BIT}. */
public static final Species<Long> SPECIES_64 = new LongSpecies(Shape.S_64_BIT, Long64Vector.class, Long64Vector.Long64Mask.class,
! Long64Vector::new, Long64Vector.Long64Mask::new,
! Long64Vector.Long64Shuffle::new, Long64Vector.Long64Shuffle::new);
! /** Species representing {@link LongVector}s of {@link Shape#S_128_BIT Shape.S_128_BIT}. */
public static final Species<Long> SPECIES_128 = new LongSpecies(Shape.S_128_BIT, Long128Vector.class, Long128Vector.Long128Mask.class,
! Long128Vector::new, Long128Vector.Long128Mask::new,
! Long128Vector.Long128Shuffle::new, Long128Vector.Long128Shuffle::new);
! /** Species representing {@link LongVector}s of {@link Shape#S_256_BIT Shape.S_256_BIT}. */
public static final Species<Long> SPECIES_256 = new LongSpecies(Shape.S_256_BIT, Long256Vector.class, Long256Vector.Long256Mask.class,
! Long256Vector::new, Long256Vector.Long256Mask::new,
! Long256Vector.Long256Shuffle::new, Long256Vector.Long256Shuffle::new);
! /** Species representing {@link LongVector}s of {@link Shape#S_512_BIT Shape.S_512_BIT}. */
public static final Species<Long> SPECIES_512 = new LongSpecies(Shape.S_512_BIT, Long512Vector.class, Long512Vector.Long512Mask.class,
! Long512Vector::new, Long512Vector.Long512Mask::new,
! Long512Vector.Long512Shuffle::new, Long512Vector.Long512Shuffle::new);
! /** Species representing {@link LongVector}s of {@link Shape#S_Max_BIT Shape.S_Max_BIT}. */
public static final Species<Long> SPECIES_MAX = new LongSpecies(Shape.S_Max_BIT, LongMaxVector.class, LongMaxVector.LongMaxMask.class,
! LongMaxVector::new, LongMaxVector.LongMaxMask::new,
! LongMaxVector.LongMaxShuffle::new, LongMaxVector.LongMaxShuffle::new);
/**
* Preferred species for {@link LongVector}s.
* A preferred species is a species of maximal bit size for the platform.
*/
< prev index next >