168 lines
6.3 KiB
Java
168 lines
6.3 KiB
Java
/*
|
|
* Copyright (c) 2021, Huawei Technologies Co., Ltd. All rights reserved.
|
|
* Copyright (c) 2023, Oracle and/or its affiliates. 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 jdk.incubator.vector.ByteVector;
|
|
import jdk.incubator.vector.IntVector;
|
|
import jdk.incubator.vector.ShortVector;
|
|
import jdk.incubator.vector.VectorSpecies;
|
|
import jdk.incubator.vector.VectorShuffle;
|
|
|
|
/*
|
|
* @test
|
|
* @bug 8265907
|
|
* @modules jdk.incubator.vector
|
|
* @run main/othervm -XX:CompileThresholdScaling=0.3 -XX:-TieredCompilation compiler.vectorapi.TestVectorShuffleIota
|
|
*/
|
|
|
|
public class TestVectorShuffleIota {
|
|
static final VectorSpecies<Integer> SPECIESi = IntVector.SPECIES_128;
|
|
static final VectorSpecies<Short> SPECIESs = ShortVector.SPECIES_128;
|
|
static final VectorSpecies<Byte> SPECIESb = ByteVector.SPECIES_128;
|
|
|
|
static final int INVOC_COUNT = 5000;
|
|
|
|
static int[] ai = {87, 65, 78, 71};
|
|
|
|
static long expected_value(VectorSpecies<?> SPECIES, int start, int step, boolean wrap) {
|
|
long res = 0;
|
|
int lanesM1 = SPECIES.length() - 1;
|
|
if (wrap) {
|
|
res = (lanesM1 & (start + step * lanesM1));
|
|
} else {
|
|
int effective_index = start + step * lanesM1;
|
|
int wrapped_effective_index = effective_index & lanesM1;
|
|
res = (effective_index == wrapped_effective_index ?
|
|
wrapped_effective_index :
|
|
-SPECIES.length() + wrapped_effective_index);
|
|
}
|
|
return res;
|
|
}
|
|
|
|
static void validateTests(long actual, VectorSpecies<?> SPECIES, int start, int step, boolean wrap) {
|
|
long expected = expected_value(SPECIES, start, step, wrap);
|
|
if (actual != expected) {
|
|
throw new AssertionError("Result Mismatch!, actual = " + actual + " expected = " + expected);
|
|
}
|
|
}
|
|
|
|
static void testShuffleIotaB128(int start, int step, boolean wrap) {
|
|
long res = SPECIESb.iotaShuffle(start, step, wrap)
|
|
.laneSource(SPECIESb.length()-1);
|
|
validateTests(res, SPECIESb, start, step, wrap);
|
|
}
|
|
|
|
static void testShuffleIotaS128(int start, int step, boolean wrap) {
|
|
long res = SPECIESs.iotaShuffle(start, step, wrap)
|
|
.laneSource(SPECIESs.length()-1);
|
|
validateTests(res, SPECIESs, start, step, wrap);
|
|
}
|
|
|
|
static void testShuffleIotaI128(int start, int step, boolean wrap) {
|
|
long res = SPECIESi.iotaShuffle(start, step, wrap)
|
|
.laneSource(SPECIESi.length()-1);
|
|
validateTests(res, SPECIESi, start, step, wrap);
|
|
}
|
|
|
|
static void testShuffleIotaConst0B128() {
|
|
long res = SPECIESb.iotaShuffle(-32, 1, false)
|
|
.laneSource(SPECIESb.length()-1);
|
|
validateTests(res, SPECIESb, -32, 1, false);
|
|
}
|
|
|
|
static void testShuffleIotaConst0S128() {
|
|
long res = SPECIESs.iotaShuffle(-32, 1, false)
|
|
.laneSource(SPECIESs.length()-1);
|
|
validateTests(res, SPECIESs, -32, 1, false);
|
|
}
|
|
|
|
static void testShuffleIotaConst0I128() {
|
|
long res = SPECIESi.iotaShuffle(-32, 1, false)
|
|
.laneSource(SPECIESi.length()-1);
|
|
validateTests(res, SPECIESi, -32, 1, false);
|
|
}
|
|
|
|
static void testShuffleIotaConst1B128() {
|
|
long res = SPECIESb.iotaShuffle(-32, 1, true)
|
|
.laneSource(SPECIESb.length()-1);
|
|
validateTests(res, SPECIESb, -32, 1, true);
|
|
}
|
|
|
|
static void testShuffleIotaConst1S128() {
|
|
long res = SPECIESs.iotaShuffle(-32, 1, true)
|
|
.laneSource(SPECIESs.length()-1);
|
|
validateTests(res, SPECIESs, -32, 1, true);
|
|
}
|
|
|
|
static void testShuffleIotaConst1I128() {
|
|
long res = SPECIESi.iotaShuffle(-32, 1, true)
|
|
.laneSource(SPECIESi.length()-1);
|
|
validateTests(res, SPECIESi, -32, 1, true);
|
|
}
|
|
|
|
static void testShuffleI() {
|
|
IntVector iv = (IntVector) VectorShuffle.iota(SPECIESi, 0, 2, false).toVector();
|
|
iv.intoArray(ai, 0);
|
|
}
|
|
|
|
public static void main(String[] args) {
|
|
for (int i = 0; i < INVOC_COUNT; i++) {
|
|
testShuffleI();
|
|
}
|
|
for (int i = 0; i < ai.length; i++) {
|
|
System.out.print(ai[i] + ", ");
|
|
}
|
|
for (int i = 0; i < INVOC_COUNT; i++) {
|
|
testShuffleIotaI128(128, 1, true);
|
|
testShuffleIotaI128(128, 1, false);
|
|
testShuffleIotaI128(-128, 1, true);
|
|
testShuffleIotaI128(-128, 1, false);
|
|
testShuffleIotaI128(1, 1, true);
|
|
testShuffleIotaI128(1, 1, false);
|
|
|
|
testShuffleIotaS128(128, 1, true);
|
|
testShuffleIotaS128(128, 1, false);
|
|
testShuffleIotaS128(-128, 1, true);
|
|
testShuffleIotaS128(-128, 1, false);
|
|
testShuffleIotaS128(1, 1, true);
|
|
testShuffleIotaS128(1, 1, false);
|
|
|
|
testShuffleIotaB128(128, 1, true);
|
|
testShuffleIotaB128(128, 1, false);
|
|
testShuffleIotaB128(-128, 1, true);
|
|
testShuffleIotaB128(-128, 1, false);
|
|
testShuffleIotaB128(1, 1, true);
|
|
testShuffleIotaB128(1, 1, false);
|
|
testShuffleIotaConst0B128();
|
|
testShuffleIotaConst0S128();
|
|
testShuffleIotaConst0I128();
|
|
testShuffleIotaConst1B128();
|
|
testShuffleIotaConst1S128();
|
|
testShuffleIotaConst1I128();
|
|
}
|
|
System.out.println();
|
|
}
|
|
}
|