jdk-24/test/hotspot/jtreg/compiler/vectorapi/VectorCastShape128Test.java
2022-03-30 09:47:20 +00:00

565 lines
20 KiB
Java

/*
* Copyright (c) 2021, 2022, Arm Limited. All rights reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* This code is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 only, as
* published by the Free Software Foundation.
*
* This code is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
* version 2 for more details (a copy is included in the LICENSE file that
* accompanied this code).
*
* You should have received a copy of the GNU General Public License version
* 2 along with this work; if not, write to the Free Software Foundation,
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
*
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
* or visit www.oracle.com if you need additional information or have any
* questions.
*/
package compiler.vectorapi;
import java.util.Random;
import jdk.test.lib.Utils;
import jdk.incubator.vector.ByteVector;
import jdk.incubator.vector.DoubleVector;
import jdk.incubator.vector.FloatVector;
import jdk.incubator.vector.IntVector;
import jdk.incubator.vector.LongVector;
import jdk.incubator.vector.ShortVector;
import jdk.incubator.vector.VectorSpecies;
import org.testng.Assert;
import org.testng.annotations.Test;
/**
* @test
* @bug 8268966
* @key randomness
* @summary AArch64: 'bad AD file' in some vector conversion tests
* @library /test/lib
* @modules jdk.incubator.vector
* @run testng/othervm -XX:-TieredCompilation -XX:CompileThreshold=100 compiler.vectorapi.VectorCastShape128Test
*/
public class VectorCastShape128Test {
private static final VectorSpecies<Long> lspec = LongVector.SPECIES_128;
private static final VectorSpecies<Integer> ispec = IntVector.SPECIES_128;
private static final VectorSpecies<Short> sspec = ShortVector.SPECIES_128;
private static final VectorSpecies<Byte> bspec = ByteVector.SPECIES_128;
private static final VectorSpecies<Float> fspec = FloatVector.SPECIES_128;
private static final VectorSpecies<Double> dspec = DoubleVector.SPECIES_128;
private static final int NUM_ITER = 10000;
private static final int LENGTH = 1024;
private static int[] ia;
private static int[] ib;
private static byte[] ba;
private static byte[] bb;
private static short[] sa;
private static short[] sb;
private static long[] la;
private static long[] lb;
private static double[] da;
private static double[] db;
private static float[] fa;
private static float[] fb;
public static float [] fspecial = {
0.0f,
-0.0f,
Float.MAX_VALUE,
Float.MIN_VALUE,
-Float.MAX_VALUE,
-Float.MIN_VALUE,
Float.NaN,
Float.POSITIVE_INFINITY,
Float.NEGATIVE_INFINITY,
Integer.MAX_VALUE,
Integer.MIN_VALUE,
Long.MAX_VALUE,
Long.MIN_VALUE,
};
public static double [] dspecial = {
0.0,
-0.0,
Double.MAX_VALUE,
Double.MIN_VALUE,
-Double.MAX_VALUE,
-Double.MIN_VALUE,
Double.NaN,
Double.POSITIVE_INFINITY,
Double.NEGATIVE_INFINITY,
Integer.MAX_VALUE,
Integer.MIN_VALUE,
Long.MIN_VALUE,
Long.MAX_VALUE,
};
public static int [] ispecial = {
0,
Integer.MAX_VALUE,
Integer.MIN_VALUE,
};
public static long [] lspecial = {
0,
Long.MAX_VALUE,
Long.MIN_VALUE,
// A special value to make sure correct rounding of
// conversion from long to float. See: JDK-8282764.
0x561a524000000001L,
};
private static void initialize() {
ia = new int[LENGTH];
ib = new int[LENGTH];
la = new long[LENGTH];
lb = new long[LENGTH];
sa = new short[LENGTH];
sb = new short[LENGTH];
ba = new byte[LENGTH];
bb = new byte[LENGTH];
fa = new float[LENGTH];
fb = new float[LENGTH];
da = new double[LENGTH];
db = new double[LENGTH];
Random r = Utils.getRandomInstance();
for (int i = 0; i < LENGTH; i++) {
ia[i] = r.nextInt();
la[i] = r.nextLong();
sa[i] = (short) r.nextInt();
ba[i] = (byte) r.nextInt();
fa[i] = ia[i] + r.nextFloat();
da[i] = la[i] + r.nextDouble();
}
// Replicate to make sure the values get tested, as some elements may be
// ignored for some vector conversions.
for (int i = 0; i < 4; i++) {
System.arraycopy(ispecial, 0, ia, ispecial.length * i, ispecial.length);
System.arraycopy(lspecial, 0, la, lspecial.length * i, lspecial.length);
System.arraycopy(fspecial, 0, fa, fspecial.length * i, fspecial.length);
System.arraycopy(dspecial, 0, da, dspecial.length * i, dspecial.length);
}
}
private static void testDoubleToByte() {
for (int i = 0; i < dspec.loopBound(LENGTH); i += dspec.length()) {
DoubleVector va = DoubleVector.fromArray(dspec, da, i);
ByteVector vb = (ByteVector) va.castShape(bspec, 0);
vb.intoArray(bb, 0);
for (int j = 0; j < Math.min(dspec.length(), bspec.length()); j++) {
Assert.assertEquals(bb[j], (byte) da[j + i]);
}
}
}
private static void testDoubleToShort() {
for (int i = 0; i < dspec.loopBound(LENGTH); i += dspec.length()) {
DoubleVector va = DoubleVector.fromArray(dspec, da, i);
ShortVector vb = (ShortVector) va.castShape(sspec, 0);
vb.intoArray(sb, 0);
for (int j = 0; j < Math.min(dspec.length(), sspec.length()); j++) {
Assert.assertEquals(sb[j], (short) da[j + i]);
}
}
}
private static void testDoubleToInt() {
for (int i = 0; i < dspec.loopBound(LENGTH); i += dspec.length()) {
DoubleVector va = DoubleVector.fromArray(dspec, da, i);
IntVector vb = (IntVector) va.castShape(ispec, 0);
vb.intoArray(ib, 0);
for (int j = 0; j < Math.min(dspec.length(), ispec.length()); j++) {
Assert.assertEquals(ib[j], (int) da[j + i]);
}
}
}
private static void testDoubleToLong() {
for (int i = 0; i < dspec.loopBound(LENGTH); i += dspec.length()) {
DoubleVector va = DoubleVector.fromArray(dspec, da, i);
LongVector vb = (LongVector) va.castShape(lspec, 0);
vb.intoArray(lb, 0);
for (int j = 0; j < Math.min(dspec.length(), lspec.length()); j++) {
Assert.assertEquals(lb[j], (long) da[j + i]);
}
}
}
private static void testDoubleToFloat() {
for (int i = 0; i < dspec.loopBound(LENGTH); i += dspec.length()) {
DoubleVector va = DoubleVector.fromArray(dspec, da, i);
FloatVector vb = (FloatVector) va.castShape(fspec, 0);
vb.intoArray(fb, 0);
for (int j = 0; j < Math.min(dspec.length(), fspec.length()); j++) {
Assert.assertEquals(fb[j], (float) da[j + i]);
}
}
}
private static void testFloatToByte() {
for (int i = 0; i < fspec.loopBound(LENGTH); i += fspec.length()) {
FloatVector va = FloatVector.fromArray(fspec, fa, i);
ByteVector vb = (ByteVector) va.castShape(bspec, 0);
vb.intoArray(bb, 0);
for (int j = 0; j < Math.min(fspec.length(), bspec.length()); j++) {
Assert.assertEquals(bb[j], (byte) fa[j + i]);
}
}
}
private static void testFloatToShort() {
for (int i = 0; i < fspec.loopBound(LENGTH); i += fspec.length()) {
FloatVector va = FloatVector.fromArray(fspec, fa, i);
ShortVector vb = (ShortVector) va.castShape(sspec, 0);
vb.intoArray(sb, 0);
for (int j = 0; j < Math.min(fspec.length(), sspec.length()); j++) {
Assert.assertEquals(sb[j], (short) fa[j + i]);
}
}
}
private static void testFloatToInt() {
for (int i = 0; i < fspec.loopBound(LENGTH); i += fspec.length()) {
FloatVector va = FloatVector.fromArray(fspec, fa, i);
IntVector vb = (IntVector) va.castShape(ispec, 0);
vb.intoArray(ib, 0);
for (int j = 0; j < Math.min(fspec.length(), ispec.length()); j++) {
Assert.assertEquals(ib[j], (int) fa[j + i]);
}
}
}
private static void testFloatToLong() {
for (int i = 0; i < fspec.loopBound(LENGTH); i += fspec.length()) {
FloatVector va = FloatVector.fromArray(fspec, fa, i);
LongVector vb = (LongVector) va.castShape(lspec, 0);
vb.intoArray(lb, 0);
for (int j = 0; j < Math.min(fspec.length(), lspec.length()); j++) {
Assert.assertEquals(lb[j], (long) fa[j + i]);
}
}
}
private static void testFloatToDouble() {
for (int i = 0; i < fspec.loopBound(LENGTH); i += fspec.length()) {
FloatVector va = FloatVector.fromArray(fspec, fa, i);
DoubleVector vb = (DoubleVector) va.castShape(dspec, 0);
vb.intoArray(db, 0);
for (int j = 0; j < Math.min(fspec.length(), dspec.length()); j++) {
Assert.assertEquals(db[j], (double) fa[j + i]);
}
}
}
private static void testIntToByte() {
for (int i = 0; i < ispec.loopBound(LENGTH); i += ispec.length()) {
IntVector va = IntVector.fromArray(ispec, ia, i);
ByteVector vb = (ByteVector) va.castShape(bspec, 0);
vb.intoArray(bb, 0);
for (int j = 0; j < Math.min(ispec.length(), bspec.length()); j++) {
Assert.assertEquals(bb[j], (byte) ia[j + i]);
}
}
}
private static void testIntToShort() {
for (int i = 0; i < ispec.loopBound(LENGTH); i += ispec.length()) {
IntVector va = IntVector.fromArray(ispec, ia, i);
ShortVector vb = (ShortVector) va.castShape(sspec, 0);
vb.intoArray(sb, 0);
for (int j = 0; j < Math.min(ispec.length(), sspec.length()); j++) {
Assert.assertEquals(sb[j], (short) ia[j + i]);
}
}
}
private static void testIntToLong() {
for (int i = 0; i < ispec.loopBound(LENGTH); i += ispec.length()) {
IntVector va = IntVector.fromArray(ispec, ia, i);
LongVector vb = (LongVector) va.castShape(lspec, 0);
vb.intoArray(lb, 0);
for (int j = 0; j < Math.min(ispec.length(), lspec.length()); j++) {
Assert.assertEquals(lb[j], (long) ia[j + i]);
}
}
}
private static void testIntToFloat() {
for (int i = 0; i < ispec.loopBound(LENGTH); i += ispec.length()) {
IntVector va = IntVector.fromArray(ispec, ia, i);
FloatVector vb = (FloatVector) va.castShape(fspec, 0);
vb.intoArray(fb, 0);
for (int j = 0; j < Math.min(ispec.length(), fspec.length()); j++) {
Assert.assertEquals(fb[j], (float) ia[j + i]);
}
}
}
private static void testIntToDouble() {
for (int i = 0; i < ispec.loopBound(LENGTH); i += ispec.length()) {
IntVector va = IntVector.fromArray(ispec, ia, i);
DoubleVector vb = (DoubleVector) va.castShape(dspec, 0);
vb.intoArray(db, 0);
for (int j = 0; j < Math.min(ispec.length(), dspec.length()); j++) {
Assert.assertEquals(db[j], (double) ia[j + i]);
}
}
}
private static void testLongToByte() {
for (int i = 0; i < lspec.loopBound(LENGTH); i += lspec.length()) {
LongVector va = LongVector.fromArray(lspec, la, i);
ByteVector vb = (ByteVector) va.castShape(bspec, 0);
vb.intoArray(bb, 0);
for (int j = 0; j < Math.min(lspec.length(), bspec.length()); j++) {
Assert.assertEquals(bb[j], (byte) la[j + i]);
}
}
}
private static void testLongToShort() {
for (int i = 0; i < lspec.loopBound(LENGTH); i += lspec.length()) {
LongVector va = LongVector.fromArray(lspec, la, i);
ShortVector vb = (ShortVector) va.castShape(sspec, 0);
vb.intoArray(sb, 0);
for (int j = 0; j < Math.min(lspec.length(), sspec.length()); j++) {
Assert.assertEquals(sb[j], (short) la[j + i]);
}
}
}
private static void testLongToInt() {
for (int i = 0; i < lspec.loopBound(LENGTH); i += lspec.length()) {
LongVector va = LongVector.fromArray(lspec, la, i);
IntVector vb = (IntVector) va.castShape(ispec, 0);
vb.intoArray(ib, 0);
for (int j = 0; j < Math.min(lspec.length(), ispec.length()); j++) {
Assert.assertEquals(ib[j], (int) la[j + i]);
}
}
}
private static void testLongToFloat() {
for (int i = 0; i < lspec.loopBound(LENGTH); i += lspec.length()) {
LongVector va = LongVector.fromArray(lspec, la, i);
FloatVector vb = (FloatVector) va.castShape(fspec, 0);
vb.intoArray(fb, 0);
for (int j = 0; j < Math.min(lspec.length(), fspec.length()); j++) {
Assert.assertEquals(fb[j], (float) la[j + i]);
}
}
}
private static void testLongToDouble() {
for (int i = 0; i < lspec.loopBound(LENGTH); i += lspec.length()) {
LongVector va = LongVector.fromArray(lspec, la, i);
DoubleVector vb = (DoubleVector) va.castShape(dspec, 0);
vb.intoArray(db, 0);
for (int j = 0; j < Math.min(lspec.length(), dspec.length()); j++) {
Assert.assertEquals(db[j], (double) la[j + i]);
}
}
}
private static void testShortToByte() {
for (int i = 0; i < sspec.loopBound(LENGTH); i += sspec.length()) {
ShortVector va = ShortVector.fromArray(sspec, sa, i);
ByteVector vb = (ByteVector) va.castShape(bspec, 0);
vb.intoArray(bb, 0);
for (int j = 0; j < Math.min(sspec.length(), bspec.length()); j++) {
Assert.assertEquals(bb[j], (byte) sa[j + i]);
}
}
}
private static void testShortToInt() {
for (int i = 0; i < sspec.loopBound(LENGTH); i += sspec.length()) {
ShortVector va = ShortVector.fromArray(sspec, sa, i);
IntVector vb = (IntVector) va.castShape(ispec, 0);
vb.intoArray(ib, 0);
for (int j = 0; j < Math.min(sspec.length(), ispec.length()); j++) {
Assert.assertEquals(ib[j], (int) sa[j + i]);
}
}
}
private static void testShortToLong() {
for (int i = 0; i < sspec.loopBound(LENGTH); i += sspec.length()) {
ShortVector va = ShortVector.fromArray(sspec, sa, i);
LongVector vb = (LongVector) va.castShape(lspec, 0);
vb.intoArray(lb, 0);
for (int j = 0; j < Math.min(sspec.length(), lspec.length()); j++) {
Assert.assertEquals(lb[j], (long) sa[j + i]);
}
}
}
private static void testShortToFloat() {
for (int i = 0; i < sspec.loopBound(LENGTH); i += sspec.length()) {
ShortVector va = ShortVector.fromArray(sspec, sa, i);
FloatVector vb = (FloatVector) va.castShape(fspec, 0);
vb.intoArray(fb, 0);
for (int j = 0; j < Math.min(sspec.length(), fspec.length()); j++) {
Assert.assertEquals(fb[j], (float) sa[j + i]);
}
}
}
private static void testShortToDouble() {
for (int i = 0; i < sspec.loopBound(LENGTH); i += sspec.length()) {
ShortVector va = ShortVector.fromArray(sspec, sa, i);
DoubleVector vb = (DoubleVector) va.castShape(dspec, 0);
vb.intoArray(db, 0);
for (int j = 0; j < Math.min(sspec.length(), dspec.length()); j++) {
Assert.assertEquals(db[j], (double) sa[j + i]);
}
}
}
private static void testByteToShort() {
for (int i = 0; i < bspec.loopBound(LENGTH); i += bspec.length()) {
ByteVector va = ByteVector.fromArray(bspec, ba, i);
ShortVector vb = (ShortVector) va.castShape(sspec, 0);
vb.intoArray(sb, 0);
for (int j = 0; j < Math.min(bspec.length(), sspec.length()); j++) {
Assert.assertEquals(sb[j], (short) ba[j + i]);
}
}
}
private static void testByteToInt() {
for (int i = 0; i < bspec.loopBound(LENGTH); i += bspec.length()) {
ByteVector va = ByteVector.fromArray(bspec, ba, i);
IntVector vb = (IntVector) va.castShape(ispec, 0);
vb.intoArray(ib, 0);
for (int j = 0; j < Math.min(bspec.length(), ispec.length()); j++) {
Assert.assertEquals(ib[j], (int) ba[j + i]);
}
}
}
private static void testByteToLong() {
for (int i = 0; i < bspec.loopBound(LENGTH); i += bspec.length()) {
ByteVector va = ByteVector.fromArray(bspec, ba, i);
LongVector vb = (LongVector) va.castShape(lspec, 0);
vb.intoArray(lb, 0);
for (int j = 0; j < Math.min(bspec.length(), lspec.length()); j++) {
Assert.assertEquals(lb[j], (long) ba[j + i]);
}
}
}
private static void testByteToFloat() {
for (int i = 0; i < bspec.loopBound(LENGTH); i += bspec.length()) {
ByteVector va = ByteVector.fromArray(bspec, ba, i);
FloatVector vb = (FloatVector) va.castShape(fspec, 0);
vb.intoArray(fb, 0);
for (int j = 0; j < Math.min(bspec.length(), fspec.length()); j++) {
Assert.assertEquals(fb[j], (float) ba[j + i]);
}
}
}
private static void testByteToDouble() {
for (int i = 0; i < bspec.loopBound(LENGTH); i += bspec.length()) {
ByteVector va = ByteVector.fromArray(bspec, ba, i);
DoubleVector vb = (DoubleVector) va.castShape(dspec, 0);
vb.intoArray(db, 0);
for (int j = 0; j < Math.min(bspec.length(), dspec.length()); j++) {
Assert.assertEquals(db[j], (double) ba[j + i]);
}
}
}
@Test
public void testCastShape128() {
initialize();
for (int i = 0; i < NUM_ITER; i++) {
testDoubleToByte();
testDoubleToShort();
testDoubleToInt();
testDoubleToLong();
testDoubleToFloat();
testFloatToByte();
testFloatToShort();
testFloatToInt();
testFloatToLong();
testFloatToDouble();
testLongToByte();
testLongToShort();
testLongToInt();
testLongToFloat();
testLongToDouble();
testIntToByte();
testIntToShort();
testIntToLong();
testIntToFloat();
testIntToDouble();
testShortToByte();
testShortToInt();
testShortToLong();
testShortToFloat();
testShortToDouble();
testByteToShort();
testByteToInt();
testByteToLong();
testByteToFloat();
testByteToDouble();
}
}
}