infernap12/jdk
1
0
Fork 0
mirror of https://github.com/openjdk/jdk.git synced 2026-01-28 12:09:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

more test cases

Browse Source
This commit is contained in:
Emanuel Peter 2025-12-08 16:16:03 +01:00
parent 9e6e8f2d3f
commit 2e7824226f
2 changed files with 161 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

83
test/hotspot/jtreg/compiler/vectorization/TestVectorAlgorithms.java
View File

@ -68,6 +68,8 @@ public class TestVectorAlgorithms {
public static void main(String[] args) {
TestFramework framework = new TestFramework();
// TODO: run with and without SuperWord, and also some intrinsics should be disabled in a run.
// make sure that all those flags are also mentioned in the JMH.
framework.addFlags("--add-modules=jdk.incubator.vector", "-XX:CompileCommand=inline,*VectorAlgorithmsImpl::*");
framework.start();
}
@ -81,6 +83,23 @@ public class TestVectorAlgorithms {
// and run.
// An alternative to cloning is to use different return arrays for
// different implementations of the same group, e.g. rI1, rI2, ...
testGroups.put("fillI", new HashMap<String,TestFunction>());
testGroups.get("fillI").put("fillI_loop", () -> { return fillI_loop(rI1); });
testGroups.get("fillI").put("fillI_VectorAPI", () -> { return fillI_VectorAPI(rI1); });
testGroups.put("iotaI", new HashMap<String,TestFunction>());
testGroups.get("iotaI").put("iotaI_loop", () -> { return iotaI_loop(rI1); });
testGroups.get("iotaI").put("iotaI_VectorAPI", () -> { return iotaI_VectorAPI(rI1); });
testGroups.put("copyI", new HashMap<String,TestFunction>());
testGroups.get("copyI").put("copyI_loop", () -> { return copyI_loop(aI, rI1); });
testGroups.get("copyI").put("copyI_VectorAPI", () -> { return copyI_VectorAPI(aI, rI1); });
testGroups.put("mapI", new HashMap<String,TestFunction>());
testGroups.get("mapI").put("mapI_loop", () -> { return mapI_loop(aI, rI1); });
testGroups.get("mapI").put("mapI_VectorAPI", () -> { return mapI_VectorAPI(aI, rI1); });
testGroups.put("reduceAddI", new HashMap<String,TestFunction>());
testGroups.get("reduceAddI").put("reduceAddI_loop", () -> { return reduceAddI_loop(aI); });
testGroups.get("reduceAddI").put("reduceAddI_reassociate", () -> { return reduceAddI_reassociate(aI); });
@ -102,7 +121,15 @@ public class TestVectorAlgorithms {
}
@Warmup(100)
@Run(test = {"reduceAddI_loop",
@Run(test = {"fillI_loop",
"fillI_VectorAPI",
"iotaI_loop",
"iotaI_VectorAPI",
"copyI_loop",
"copyI_VectorAPI",
"mapI_loop",
"mapI_VectorAPI",
"reduceAddI_loop",
"reduceAddI_reassociate",
"reduceAddI_VectorAPI_naive",
"reduceAddI_VectorAPI_reduction_after_loop",
@ -211,4 +238,58 @@ public class TestVectorAlgorithms {
public Object reverse_VectorAPI(int[] a, int[] r) {
return VectorAlgorithmsImpl.reverse_VectorAPI(a, r);
}
@Test
//@IR(counts = {IRNode.STORE_VECTOR, "> 0"},
// applyIfCPUFeatureOr = {"sse4.1", "true", "asimd", "true"})
// TODO:
// 250 CallLeafNoFP === 208 1 7 8 9 (248 131 251 1 ) [[ 253 254 ]] # arrayof_jint_fill void ...
public Object fillI_loop(int[] r) {
return VectorAlgorithmsImpl.fillI_loop(r);
}
@Test
public Object fillI_VectorAPI(int[] r) {
return VectorAlgorithmsImpl.fillI_VectorAPI(r);
}
@Test
@IR(counts = {IRNode.POPULATE_INDEX, "> 0",
IRNode.STORE_VECTOR, "> 0"},
applyIfCPUFeatureOr = {"sse4.1", "true", "asimd", "true"})
public Object iotaI_loop(int[] r) {
return VectorAlgorithmsImpl.iotaI_loop(r);
}
@Test
public Object iotaI_VectorAPI(int[] r) {
return VectorAlgorithmsImpl.iotaI_VectorAPI(r);
}
@Test
@IR(counts = {IRNode.LOAD_VECTOR_I, "> 0",
IRNode.STORE_VECTOR, "> 0"},
applyIfCPUFeatureOr = {"sse4.1", "true", "asimd", "true"})
public Object copyI_loop(int[] a, int[] r) {
return VectorAlgorithmsImpl.copyI_loop(a, r);
}
@Test
public Object copyI_VectorAPI(int[] a, int[] r) {
return VectorAlgorithmsImpl.copyI_VectorAPI(a, r);
}
@Test
@IR(counts = {IRNode.LOAD_VECTOR_I, "> 0",
IRNode.MUL_VI, "> 0",
IRNode.STORE_VECTOR, "> 0"},
applyIfCPUFeatureOr = {"sse4.1", "true", "asimd", "true"})
public Object mapI_loop(int[] a, int[] r) {
return VectorAlgorithmsImpl.mapI_loop(a, r);
}
@Test
public Object mapI_VectorAPI(int[] a, int[] r) {
return VectorAlgorithmsImpl.mapI_VectorAPI(a, r);
}
}

79
test/hotspot/jtreg/compiler/vectorization/VectorAlgorithmsImpl.java
View File

@ -34,6 +34,84 @@ public class VectorAlgorithmsImpl {
private static final VectorSpecies<Integer> SPECIES_I = IntVector.SPECIES_PREFERRED;
private static final VectorSpecies<Integer> SPECIES_I512 = IntVector.SPECIES_512;
public static Object fillI_loop(int[] r) {
for (int i = 0; i < r.length; i++) {
r[i] = 42;
}
return r;
}
public static Object fillI_VectorAPI(int[] r) {
var v = IntVector.broadcast(SPECIES_I512, 42);
int i = 0;
for (; i < SPECIES_I.loopBound(r.length); i += SPECIES_I.length()) {
v.intoArray(r, i);
}
for (; i < r.length; i++) {
r[i] = 42;
}
return r;
}
public static Object iotaI_loop(int[] r) {
for (int i = 0; i < r.length; i++) {
r[i] = i;
}
return r;
}
public static Object iotaI_VectorAPI(int[] r) {
var iota = IntVector.broadcast(SPECIES_I, 0).addIndex(1);
int i = 0;
for (; i < SPECIES_I.loopBound(r.length); i += SPECIES_I.length()) {
iota.intoArray(r, i);
iota = iota.add(SPECIES_I.length());
}
for (; i < r.length; i++) {
r[i] = i;
}
return r;
}
public static Object copyI_loop(int[] a, int[] r) {
for (int i = 0; i < a.length; i++) {
r[i] = a[i];
}
return r;
}
public static Object copyI_VectorAPI(int[] a, int[] r) {
int i = 0;
for (; i < SPECIES_I.loopBound(r.length); i += SPECIES_I.length()) {
IntVector v = IntVector.fromArray(SPECIES_I512, a, i);
v.intoArray(r, i);
}
for (; i < r.length; i++) {
r[i] = a[i];
}
return r;
}
public static Object mapI_loop(int[] a, int[] r) {
for (int i = 0; i < a.length; i++) {
r[i] = a[i] * 42;
}
return r;
}
public static Object mapI_VectorAPI(int[] a, int[] r) {
int i = 0;
for (; i < SPECIES_I.loopBound(r.length); i += SPECIES_I.length()) {
IntVector v = IntVector.fromArray(SPECIES_I512, a, i);
v = v.mul(42);
v.intoArray(r, i);
}
for (; i < r.length; i++) {
r[i] = a[i];
}
return r;
}
public static int reduceAddI_loop(int[] a) {
int sum = 0;
for (int i = 0; i < a.length; i++) {
@ -216,4 +294,5 @@ public class VectorAlgorithmsImpl {
}
return r;
}
}