diff options
Diffstat (limited to 'src/main/java/org/linaro/benchmarks/vector/TestSIMDLd4St4.java')
-rw-r--r-- | src/main/java/org/linaro/benchmarks/vector/TestSIMDLd4St4.java | 130 |
1 files changed, 130 insertions, 0 deletions
diff --git a/src/main/java/org/linaro/benchmarks/vector/TestSIMDLd4St4.java b/src/main/java/org/linaro/benchmarks/vector/TestSIMDLd4St4.java new file mode 100644 index 0000000..e53075c --- /dev/null +++ b/src/main/java/org/linaro/benchmarks/vector/TestSIMDLd4St4.java @@ -0,0 +1,130 @@ +/* + * Copyright (C) 2016 Linaro Limited. All rights received. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + * + */ +package org.linaro.benchmarks.vector; + +import org.openjdk.jmh.annotations.*; +import java.util.concurrent.TimeUnit; + +@BenchmarkMode(Mode.AverageTime) +@OutputTimeUnit(TimeUnit.MICROSECONDS) +@State(Scope.Benchmark) + +public class TestSIMDLd4St4 { + static final int VECT_LENGTH = 1024; + static final int COUNT = 128; + static final int DIM = 4; + static final int LENGTH = VECT_LENGTH * DIM; + static int [] in = new int[LENGTH]; + static int [] out = new int[LENGTH]; + static int [] c = new int[DIM]; + static short [] sin = new short[LENGTH]; + static short [] sout = new short[LENGTH]; + static short [] sc = new short[DIM]; + + public static void vect4DInit() { + int i; + for (i = 0; i < LENGTH; i++) { + in[i] = i + 3; + sin[i] = (short)(i + 3); + } + for (i = 0; i < DIM; i++) { + c[i] = i + 1; + sc[i] = (short)(i + 1); + } + } + + public static void vect4DAddInt( + int[] out, + int[] in, + int[] c) { + int c0 = c[0]; + int c1 = c[1]; + int c2 = c[2]; + int c3 = c[3]; + for (int i = 0; i < VECT_LENGTH; i++) { + out[i*DIM] = in[i*DIM] + c0; + out[i*DIM + 1] = in[i*DIM + 1] + c1; + out[i*DIM + 2] = in[i*DIM + 2] + c2; + out[i*DIM + 3] = in[i*DIM + 3] + c3; + } + } + + public static void vect4DMulInt( + int[] out, + int[] in, + int[] c) { + int c0 = c[0]; + int c1 = c[1]; + int c2 = c[2]; + int c3 = c[3]; + for (int i = 0; i < VECT_LENGTH; i++) { + out[i*DIM] = in[i*DIM] * c0; + out[i*DIM + 1] = in[i*DIM + 1] * c1; + out[i*DIM + 2] = in[i*DIM + 2] * c2; + out[i*DIM + 3] = in[i*DIM + 3] * c3; + } + } + + public static void vect4DAddShort( + short[] out, + short[] in, + short[] c) { + short c0 = c[0]; + short c1 = c[1]; + short c2 = c[2]; + short c3 = c[3]; + for (int i = 0; i < VECT_LENGTH; i++) { + out[i*DIM] = (short)(in[i*DIM] + c0); + out[i*DIM + 1] = (short)(in[i*DIM + 1] + c1); + out[i*DIM + 2] = (short)(in[i*DIM + 2] + c2); + out[i*DIM + 3] = (short)(in[i*DIM + 3] + c3); + } + } + + public static void vect4DMulShort( + short[] out, + short[] in, + short[] c) { + short c0 = c[0]; + short c1 = c[1]; + short c2 = c[2]; + short c3 = c[3]; + for (int i = 0; i < VECT_LENGTH; i++) { + out[i*DIM] = (short)(in[i*DIM] * c0); + out[i*DIM + 1] = (short)(in[i*DIM + 1] * c1); + out[i*DIM + 2] = (short)(in[i*DIM + 2] * c2); + out[i*DIM + 3] = (short)(in[i*DIM + 3] * c3); + } + } + + @Setup + public void setup() + { + vect4DInit(); + } + + @Benchmark + public void testVect4D() { + for (int i = 0; i < COUNT; i++) { + vect4DAddInt(out, in, c); + vect4DMulInt(out, in, c); + vect4DAddShort(sout, sin, sc); + vect4DMulShort(sout, sin, sc); + } + } + +} |