17 lines
857 B
Java
17 lines
857 B
Java
import jdk.incubator.vector.*;
|
|
|
|
/** JEP 448 (sixth incubator): needs --add-modules jdk.incubator.vector. Result must equal the scalar loop. */
|
|
public class VectorDemo {
|
|
static final VectorSpecies<Integer> S = IntVector.SPECIES_PREFERRED;
|
|
public static void main(String[] args) {
|
|
int n = 1000; int[] a = new int[n], b = new int[n], c = new int[n], d = new int[n];
|
|
for (int i = 0; i < n; i++) { a[i] = i; b[i] = 2 * i; }
|
|
int i = 0;
|
|
for (; i < S.loopBound(n); i += S.length()) IntVector.fromArray(S, a, i).add(IntVector.fromArray(S, b, i)).intoArray(c, i);
|
|
for (; i < n; i++) c[i] = a[i] + b[i];
|
|
for (int k = 0; k < n; k++) d[k] = a[k] + b[k];
|
|
System.out.println("vector == scalar: " + java.util.Arrays.equals(c, d));
|
|
System.out.println("lanes >= 2 : " + (S.length() >= 2));
|
|
}
|
|
}
|